如下兩個數據框，我想計算相關系數。當兩列都用實際值完成時，它工作正常。但如果不是，則在計算相關系數時取零作為值。例如，Addison 和 Caden 的權重為 0。Jack 和 Noah 沒有權重。我想排除它們進行計算。（在嘗試中，似乎只考慮相同的長度，即自動排除 Jack 和 Noah – 是嗎？）如何只包括非零值的人進行計算？謝謝你。import pandas as pdWeight = {'Name': ["Abigail","Addison","Aiden","Amelia","Aria","Ava","Caden","Charlotte","Chloe","Elijah"], 'Weight': [10, 0, 12, 20, 25, 10, 0, 18, 16, 13]}df_wt = pd.DataFrame(Weight)Score = {'Name': ["Abigail","Addison","Aiden","Amelia","Aria","Ava","Caden","Charlotte","Chloe","Elijah", "Jack", "Noah"], 'Score': [360, 476, 345, 601, 604, 313, 539, 531, 507, 473, 450, 470]}df_sc = pd.DataFrame(Score)print df_wt.Weight.corr(df_sc.Score)
2 回答
翻閱古今
TA貢獻1780條經驗 獲得超5個贊
屏蔽和取非零值和公共索引：
df_wt.set_index('Name', inplace=True)
df_sc.set_index('Name', inplace=True)
mask = df_wt['Weight'].ne(0)
common_index = df_wt.loc[mask, :].index
df_wt.loc[common_index, 'Weight'].corr(df_sc.loc[common_index, 'Score'])
0.923425144491911
如果兩個數據幀都包含零，則：
mask1 = df_wt['Weight'].ne(0)
mask2 = df_sc['Score'].ne(0)
common_index = df_wt.loc[mask1, :].index.intersection(df_sc.loc[mask2, :].index)
df_wt.loc[common_index, 'Weight'].corr(df_sc.loc[common_index, 'Score'])
MMMHUHU
TA貢獻1834條經驗 獲得超8個贊
屏蔽和取非零值和公共索引：
df_wt.set_index('Name', inplace=True)
df_sc.set_index('Name', inplace=True)
mask = df_wt['Weight'].ne(0)
common_index = df_wt.loc[mask, :].index
df_wt.loc[common_index, 'Weight'].corr(df_sc.loc[common_index, 'Score'])
0.923425144491911
如果兩個數據幀都包含零，則：
mask1 = df_wt['Weight'].ne(0)
mask2 = df_sc['Score'].ne(0)
common_index = df_wt.loc[mask1, :].index.intersection(df_sc.loc[mask2, :].index)
df_wt.loc[common_index, 'Weight'].corr(df_sc.loc[common_index, 'Score'])
添加回答
舉報
0/150
提交
取消