怎么获取groupby()字段名
时间: 2023-08-18 20:02:25 浏览: 91
group by用法.doc
要获取使用groupby()方法进行分组的字段名,可以使用GroupBy对象的属性来获取。GroupBy对象是一个中间结果,表示按照指定字段进行分组后的数据。
下面是一个示例代码:
```python
import pandas as pd
# 创建一个包含电影数据的DataFrame
movies = pd.DataFrame({
'Title': ['Movie A', 'Movie B', 'Movie C', 'Movie A', 'Movie B'],
'Genre': ['Action', 'Comedy', 'Action', 'Drama', 'Comedy'],
'Rating': [7.5, 6.8, 8.2, 7.9, 6.5]
})
# 按照Genre列进行分组
grouped_movies = movies.groupby('Genre')
# 获取分组字段名
grouped_field = grouped_movies.grouper.names[0]
print(grouped_field)
```
输出结果为:
```
Genre
```
在上述示例中,我们使用movies.groupby('Genre')将数据按照Genre列进行分组,并将结果存储在grouped_movies变量中。然后,我们可以通过grouped_movies.grouper.names[0]来获取分组字段名,即'Genre'。
阅读全文