根据其他列中的值使用熊猫添加计算列

时间:2019-11-06 07:02:41

标签: python python-3.x pandas

我需要有关基于Col1,2和Col3的数据添加C​​ol4的帮助。如果Col3在Col1 / Col2中的所有对应值都具有相同的值,则Col4应读为“是”,否则为“否”。

[Data]

[ExpectedResult_Col4]

1 个答案:

答案 0 :(得分:2)

GroupBy.transform与唯一值计数一起使用,并按1进行比较,按numpy.where设置新值:

mask = df.groupby(['Col1','Col2'])['Col3'].transform('nunique') == 1
df['Col4'] = np.where(mask, 'yes', 'no')