按ID分组检查DataFrame同组name是否一致并添加布尔列

假设你的数据已加载到 pandas DataFrame 中，可通过以下步骤实现需求：

1. 按 ID 分组，检查每组内 name 列的唯一值数量是否为 1
2. 使用 transform 方法将分组结果映射回原 DataFrame 的每一行，生成新列 names.equal

代码实现 #

import pandas as pd

# 构造示例数据
data = {
    'title.1': ['A', 'A', '', 'B', 'C', 'C', 'C', '', ''],
    'title.2': ['A1', 'A1', 'B1', '', 'C1', 'C1', 'C1', 'D1', 'D1'],
    'name': ['fruit', 'fruit', 'fruit', 'fruit, vegetable', 'vegetable, poultry, grain', 'vegetable, poultry', 'vegetable, poultry', 'poultry', 'vegetable'],
    'ID': [1, 1, 2, 3, 4, 4, 4, 5, 5]
}
df = pd.DataFrame(data)

# 新增 names.equal 列
df['names.equal'] = df.groupby('ID')['name'].transform(lambda x: x.nunique() == 1)

print(df)

输出结果 #

内容的提问来源于stack exchange，提问作者bvecc