要获取DataFrame GroupBy对象的列名,可以使用GroupBy对象的groups
属性。该属性返回一个字典,其中键是分组的标签,值是该分组对应的行索引。通过获取字典的键,即可得到分组的列名。
以下是一个示例代码:
import pandas as pd
# 创建一个DataFrame
data = {'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'two', 'two', 'one', 'two', 'one'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80]}
df = pd.DataFrame(data)
# 按列'A'进行分组
grouped = df.groupby('A')
# 获取分组的列名
column_names = list(grouped.groups.keys())
print(column_names)
输出结果为:
['bar', 'foo']
在这个示例中,我们首先创建了一个DataFrame对象df
,然后按列'A'进行分组,得到一个GroupBy对象grouped
。通过grouped.groups.keys()
获取分组的列名,并将其转换为列表形式存储在column_names
中。最后打印column_names
即可得到分组的列名。
对于腾讯云相关产品和产品介绍链接地址,由于要求不能提及具体的云计算品牌商,因此无法提供相关链接。
领取专属 10元无门槛券
手把手带您无忧上云