在multindex熊猫数据框上应用功能

时间:2020-04-17 20:38:11

标签: python pandas

我有一个与以下类似的数据框

data = {'lat': [90,80,70],'long':[100,110,100],'temp1':[[4,3,1,1],[6,2,3,-1],[1,2,3,4]],'temp2':[[1,2,3,4],[1,2,3,4],[9,9,1,3]]}
df3 = pd.DataFrame(data)
df3 = df3.set_index(['lat','long'])

我想为每个multindex行计算“ temp1”和“ temp2”之间的相关性,为此,我定义了以下函数并使用了apply方法:

def corr_row(row):
    serie1 = row['temp1']
    serie2 = row['temp2']
    correlation = np.corrcoef(serie1,serie2)
    return correlation

df3["corr"] = df3.apply(corr_row)

但是我得到了错误:

Exception: Data must be 1-dimensional

我设法在具有一个索引的数据帧中使用apply方法,在这种情况下应如何使用?

0 个答案:

没有答案