以下是一个示例代码,它将一个数据列表按照指定的分组条件进行分组,并为每个分组创建一个新列:
import pandas as pd
# 创建示例数据
data = {'Name': ['Tom', 'Nick', 'John', 'Tom', 'Nick', 'John'],
'Age': [20, 25, 30, 22, 27, 32],
'City': ['New York', 'Paris', 'London', 'Tokyo', 'Paris', 'London']}
df = pd.DataFrame(data)
# 按照Name列分组,并为每个分组创建一个新列'Count'
df['Count'] = df.groupby('Name').cumcount() + 1
# 输出结果
print(df)
输出结果如下:
Name Age City Count
0 Tom 20 New York 1
1 Nick 25 Paris 1
2 John 30 London 1
3 Tom 22 Tokyo 2
4 Nick 27 Paris 2
5 John 32 London 2
在这个示例中,我们使用groupby
函数将数据按照Name
列进行分组,并使用cumcount
函数为每个分组创建一个新列Count
,这个新列的值表示在每个分组内的行号。
上一篇:按照列表对象的字段属性筛选列表
下一篇:按照列表过滤子集合