Question

Col1    Col2   Col3   Result

2        70     1      15
2        71     2      15
2        72     3      15
3        80     4      16    
3        81     5      16
3        82     6      16
3        2      15     16
3        3      16     16

我是熊猫新手，谁能解释一下如何获得最后一列结果以添加现有数据框？

Answer 1

对所有具有col2值的唯一行使用Series.map和DataFrame.drop_duplicates：

df['Res'] = df['Col1'].map(df.drop_duplicates('col2').set_index('Col2')['Col3'])
print (df)
   Col1  Col2  Col3  Result  Res
0     2    70     1      15   15
1     2    71     2      15   15
2     2    72     3      15   15
3     3    80     4      16   16
4     3    81     5      16   16
5     3    82     6      16   16
6     3     2    15      16   16
7     3     3    16      16   16

Answer 2

另一个选项是merge：

df.merge(df[['Col2','Col3']].rename(columns={'Col2':'Col1', 'Col3':'Res'}),
         on='Col1', how='left')

输出：

   Col1  Col2  Col3  Result  Res
0     2    70     1      15   15
1     2    71     2      15   15
2     2    72     3      15   15
3     3    80     4      16   16
4     3    81     5      16   16
5     3    82     6      16   16
6     3     2    15      16   16
7     3     3    16      16   16

我们如何应用pandas groupby获得预期的输出？

2 个答案: