如何使用pandas.DataFrame.corr仅计算指定数量列的相关系数?
解决方法
你需要先锁定数据集的前4个变量,再计算它们之间的两两相关系数,就能得到4×4的目标相关矩阵。以下是两种实用实现方式:
方法一:先截取前4列再计算相关性
直接提取前4个变量组成子数据集,再调用corr()方法,这种方式更高效,避免了对其他变量的无意义计算:
import pandas as pd df = pd.read_csv('random_data.csv') # 提取前4列的子数据集,计算内部相关系数 df.iloc[:, :4].corr()
方法二:在完整相关矩阵中同时筛选行和列
如果已经生成了全变量相关矩阵,可以通过同时截取前4行和前4列来得到结果:
import pandas as pd df = pd.read_csv('random_data.csv') # 先计算全变量相关矩阵,再取前4行×前4列的子矩阵 full_corr = df.corr() target_corr = full_corr.iloc[:4, :4]
内容的提问来源于stack exchange,提问作者rshmoney2314
相关产品推荐
相关产品推荐

