将DataFrameGroupBy转换为数据框，以便获得单个列

Question

我在以下几列中有一个数据集：

  

国家

     

GDP

     

生活满意度

我想按国家和GDP分组。我想要列表名称X中的GDP col和列表Y中的生活满意度来创建散点图。

使用此功能，我可以进行分组：

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

但是df是generic.DataFrame.GroupBy

如何扩展代码以采用X和Y中的值。

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()