应用因子分析(FA)到分组数据处理中已有的分组信息。主要步骤包括数据准备,因子分析,因子载荷和因子得分计算等。以下是Python的示例代码:
import pandas as pd from factor_analyzer import FactorAnalyzer
df_groupA = pd.read_csv('groupA.csv') df_groupB = pd.read_csv('groupB.csv')
df_all_groups = pd.concat([df_groupA, df_groupB], axis=0)
n_factors = 3 # 可根据实际需要进行修改 fa = FactorAnalyzer(n_factors, rotation='varimax', method='ml') fa.fit(df_all_groups)
loadings = pd.DataFrame(fa.loadings_, index=df_all_groups.columns) print('因子载荷矩阵:\n', loadings)
factors = pd.DataFrame(fa.transform(df_all_groups), columns=['factor_1', 'factor_2', 'factor_3']) print('因子得分:\n', factors)