以下是一个示例代码,用于按照id分组,并为每个组创建一个布尔列:
import pandas as pd
# 创建一个示例数据框
data = {'id': [1, 1, 2, 2, 3, 3],
'value': [10, 20, 30, 40, 50, 60]}
df = pd.DataFrame(data)
# 按照id分组,并为每个组创建一个布尔列
df['group_bool'] = df.groupby('id').transform(lambda x: True)
print(df)
输出结果如下:
id value group_bool
0 1 10 True
1 1 20 True
2 2 30 True
3 2 40 True
4 3 50 True
5 3 60 True
在这个示例中,我们使用groupby
函数将数据框按照'id'列进行分组。然后,我们使用transform
函数将每个分组中的元素转换为布尔值True,并将结果存储在新创建的'group_bool'列中。
下一篇:按照ID分组并更新列