如何将一列关联到多列和 return 列表？

Question

我想在一列和其他列之间创建一个相关函数，将数据框与所有列一起传递，与特定列相关并返回一个指标和相关性列表，我正在这样做。

correlations = df.corr().unstack().sort_values(ascending=True) 
correlations = pd.DataFrame(correlations).reset_index() 
correlations.columns = ['corr_matrix', 'dfbase', 'correlation'] 
correlations.query("corr_matrix == 'venda por m2' & dfbase != 'venda por m2'") 

但我想知道一种使用函数实现此功能的方法。

Answer 1

应该这样做

def get_nonself_correlation(df,self_name):
    temp = df.corr()
    temp = temp.loc[temp.index!=self_name,temp.columns==self_name]
    temp = temp.unstack().reset_index()
    temp.columns = ['corr_matrix', 'dfbase', 'correlation'] 
    return temp