добавить вычисляемый столбец, используя панды на основе значений в другом столбце

Question

Мне нужна помощь с добавлением Col4 на основе данных из Col1,2 и Col3. Если Col3 имеет одинаковые значения для всех соответствующих значений в Col1 / Col2, Col4 следует читать как «ДА», иначе «НЕТ».

[]

[]

Answer 1

Используйте GroupBy.transform с количеством уникальных значений и сравните с помощью 1, установите новые значения с помощью numpy.where:

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