python - 将 DataFrameGroupBy 转换为数据框，以便我可以获取各个列

Question

我有一个数据集以下列：

国家

国内生产总值

生活满意度

我想按国家和 GDP 分组。我希望列表中的 GDP col 名称 X 和列表 Y 中的 Life Satisfaction 来创建散点图。

使用它我可以分组：

df=dataSet.groupby(["Country","Gdp Per Capita"])

但是df是generic.DataFrame.GroupBy

如何扩展我的代码以获取 X 和 Y 中的值。

标签： pythongroup-by

Answer 1

如果我正确理解了这个问题：

df=dataSet.groupby(["Country","Gdp Per Capita"], as_index=False).mean()
X = df['Gdp Per Capita'].to_list()
Y = df['Life Satisfaction'].to_list()