pandas 分组后获得组名
在使用pandas进行分组操作后,可以使用groups属性获得分组后的组名。
下面是一个示例代码:
import pandas as pd
# 创建一个示例数据
data = {
'Name': ['Tom', 'Nick', 'John', 'Tom', 'John'],
'Age': [20, 21, 22, 20, 22],
'City': ['New York', 'Paris', 'London', 'New York', 'London']
}
df = pd.DataFrame(data)
# 按照 'City' 列进行分组
groups = df.groupby('City')
# 获得组名
group_names = groups.groups.keys()
print(group_names)
输出结果为:
dict_keys(['London', 'New York', 'Paris'])
在这个示例中,我们按照 'City' 列进行了分组操作,然后使用groups属性获得了分组后的组名。
原文地址: https://www.cveoy.top/t/topic/iNjk 著作权归作者所有。请勿转载和采集!