可以使用groupby()
函数按照条件对pandas中的列值进行分组。以下是一个示例代码:
import pandas as pd
# 创建一个示例DataFrame
data = {'Animal': ['Dog', 'Cat', 'Elephant', 'Mouse', 'Lion'],
'Color': ['Brown', 'Black', 'Gray', 'Gray', 'Yellow'],
'Size': ['Large', 'Medium', 'Large', 'Small', 'Large']}
df = pd.DataFrame(data)
# 按照'Color'列的值进行分组
grouped = df.groupby('Color')
# 打印每个分组的内容
for name, group in grouped:
print(name)
print(group)
print()
输出结果为:
Black
Animal Color Size
1 Cat Black Medium
Brown
Animal Color Size
0 Dog Brown Large
Gray
Animal Color Size
2 Elephant Gray Large
3 Mouse Gray Small
Yellow
Animal Color Size
4 Lion Yellow Large
在上述代码中,我们首先创建了一个包含动物、颜色和大小的DataFrame。然后,我们使用groupby()
函数将数据按照'Color'列的值进行分组。最后,我们使用一个循环遍历每个分组,并打印出每个分组的内容。