我有一个与以下类似的数据框
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方法,在这种情况下应如何使用?