鉴于这些数据框......:
DF = pd.DataFrame({'COL1': ['A', 'B', 'C', 'D','D','D'],
'COL2': [11032, 1960, 11400, 11355, 8, 7],
'year': ['2016', '2017', '2018', '2019', '2020', '2021']})
DF
COL1 COL2 year
0 A 11032 2016
1 B 1960 2017
2 C 11400 2018
3 D 11355 2019
4 D 8 2020
5 D 7 2021
DF2 = pd.DataFrame({'ColX': ['D'], 'ColY':['2021'], 'ColZ':[100]
DF2
ColX ColY ColZ
0 D 2021 100
如果满足以下条件:
COL1 =来自DF2的ColX
year =来自DF2的ColY
然后将COL2中的值从DF2更改为ColZ。
答案 0 :(得分:2)
您希望update
DF
使用来自DF2
的数据。
假设DF2
中的所有值对ColX
和ColY
中的给定值对都是唯一的:
DF = DF.merge(DF2.set_index(['ColX', 'ColY'])[['ColZ']],
how='left',
left_on=['COL1', 'year'],
right_index=True)
DF.COL2.update(DF.ColZ)
del DF['ColZ']
>>> DF
COL1 COL2 year
0 A 11032 2016
1 B 1960 2017
2 C 11400 2018
3 D 11355 2019
4 D 8 2020
5 D 100 2021
我将一个临时数据框(DF2.set_index(['ColX', 'ColY'])[['ColZ']]
)合并到DF中,后者添加了ColZ的所有值,其索引(ColX
和ColY
)与COL1
的值匹配和year
中的DF
。所有不匹配的值都填充NA
。
然后我使用update
从DF.ColZ中的非空值覆盖DF.COL2
中的值。
然后我删除DF [' ColZ']进行清理。
如果ColZ
与DF
中的现有列名匹配,则需要进行一些调整。
另一种解决方案如下:
DF = DF.set_index(['COL1', 'year']).update(DF2.set_index(['ColX', 'ColY']))
DF.reset_index(inplace=True)
输出与上述输出相同。