python-3.x - 从 Pandas df 中选择列/轴进行关联

Question

我有一个如下所示的 Pandas 数据框。我想建立一个相关矩阵，为一系列客户记录建立产品所有权与利润/成本/收入之间的关系。

    prod_owned_a    prod_owned_b    profit      cost    rev
0   1               0               100         75      175
1   0               1               125         100     225
2   1               0               100         75      175
3   1               1               225         175     400
4   0               1               125         100     225

理想情况下，矩阵将所有 prod_owned 沿着一个轴，利润/成本/收入沿着另一轴。我想避免在相关矩阵中包含 prod_owned_a 和 prod_owned_b 之间的相关性。

问题:如何为每个轴选择特定的列？谢谢!

Answer 1

只要列的顺序不变，就可以使用切片:

df.corr().loc[:'prod_owned_b', 'profit':]
#                profit      cost       rev
#prod_owned_a  0.176090  0.111111  0.147442
#prod_owned_b  0.616316  0.666667  0.638915

一个更强大的解决方案定位所有“prod_*”列:

prod_cols = df.columns.str.match('prod_')
df.corr().loc[prod_cols, ~prod_cols]
#                profit      cost       rev
#prod_owned_a  0.176090  0.111111  0.147442
#prod_owned_b  0.616316  0.666667  0.638915