преобразовать DataFrameGroupBy во фрейм данных, чтобы я мог получить отдельные столбцы

Question

У меня есть набор данных следующих столбцов:

Страна

ВВП

Удовлетворенность жизнью

Я хочу сгруппировать пострана и ВВП.Я хочу, чтобы ВВП col в списке с именами X и Life Satisfaction в списке 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()