[英]Create new column in a DataFrame using values from a different row
我正在尝试在 pandas dataframe 中创建一个新列,这是明年相同 ID 的分数。 请参阅下面的示例原始数据:
Year ID Score
2018 785 8.4
2018 770 -1.2
2017 733 3.2
2017 785 7.9
2018 733 3.9
如果没有下一年的数据,则应填写 NA。 所以我正在寻找的 output 将是:
Year ID Score col
2018 785 8.4 NA
2018 770 -1.2 NA
2017 733 3.2 3.9
2017 785 7.9 8.4
2018 733 3.9 NA
数据当前未排序。
如果您的数据对于所有 ID 都有连续年份(没有2016 2018
的 ID),那么您可以执行以下操作:
df['col'] = df.sort_values('Year').groupby('ID').Score.shift(-1)
Output:
Year ID Score col
0 2018 785 8.4 NaN
1 2018 770 -1.2 NaN
2 2017 733 3.2 3.9
3 2017 785 7.9 8.4
4 2018 733 3.9 NaN
如果不能保证年份是连续的,则进行合并:
df.merge(df.assign(Year=lambda x: x.Year - 1),
on=['Year', 'ID'],
suffixes = ['','_new'],
how='left')
Output:
Year ID Score Score_new
0 2018 785 8.4 NaN
1 2018 770 -1.2 NaN
2 2017 733 3.2 3.9
3 2017 785 7.9 8.4
4 2018 733 3.9 NaN
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.