根据其他数据框有条件地填充数据框

时间:2018-09-26 17:46:27

标签: python pandas dataframe

我有两个数据框。第一个数据帧df包含5分钟(返回)数据以及索引的小时和分钟

import pandas as pd
dates = pd.date_range(start='2018-01-01', end='2018-01-05', freq='5T')
df = pd.DataFrame(np.random.randn(len(dates),1), index=dates, columns=list(['Ret']))
df['Hour'] = df.index.hour
df['Minute'] = df.index.minute
tmp = df['Ret'].groupby([df['Hour'],df['Minute']]).mean().to_frame()

所以现在tmp包含一个MultiIndexed数据帧,它具有小时和分钟两个级别。我现在想做的是在df中创建一个新列,在其中我从数据框tmp中选择正确的值,然后将它们放回到df数据框中。

因此,基本上,我想在df中对df['Hour']df['Minute']进行查找并返回正确的值,从而在tmp中创建一个新列,但是我可以似乎无法正常工作。

1 个答案:

答案 0 :(得分:1)

使用merge,但要注意语法:

result = df.merge(tmp, left_on=['Hour', 'Minute'], right_index=True, how='left')

result.head()
               Ret_x  Hour  Minute     Ret_y
2018-01-01 -0.751599     0       0 -0.035756
2018-01-02 -0.650874     0       0 -0.035756
2018-01-03 -1.110701     0       0 -0.035756
2018-01-04  0.944097     0       0 -0.035756
2018-01-05  1.390296     0       0 -0.035756