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

Question

我需要有关根据 Col1、2 和 Col3 的数据添加 Col4 的帮助。如果 Col3 与 Col1/Col2 中的所有对应值具有相同的值，则 Col4 应读为“YES”，否则为“NO”。

[]

[]

Answer 1

将GroupBy.transform 与唯一值的计数一起使用并通过1 进行比较，通过numpy.where 设置新值：

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