在使用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属性获得了分组后的组名。

pandas 分组后获得组名

原文地址: https://www.cveoy.top/t/topic/iNjk 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录