Pandas计算DataFrame各列之间的相关系数
在 Pandas 中，可以使用 corr() 方法计算 DataFrame 中各列之间的相关系数。该方法会返回一个矩阵，矩阵中的每个元素都是两列之间的相关系数。下面是一个示例代码：
import pandas as pd # 创建一个 DataFrame df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}) # 计算 DataFrame 中各列之间的相关系数 corr_matrix = df.corr() # 打印相关系数矩阵 print(corr_matrix)
输出结果为：
A B C A 1.0 1.0 1.0 B 1.0 1.0 1.0 C 1.0 1.0 1.0
上述示例代码中，首先创建了一个包含三列的 DataFrame df，然后使用 corr() 方法计算 DataFrame 中各列之间的相关系数，并将结果赋值给 corr_matrix 变量。最后将结果打印出来。
相关文章