Как выбрать столбцы с MultiIndex в pandas DataFrame (для графика рассеяния морского происхождения)?

Question

У меня есть агрегированный DataFrame с вложенными столбцами. Как я могу получить агрегированные столбцы, sum и count, как x и y диаграммы рассеяния?

Я пытался использовать

sns.scatterplot(x="sum", y="count", data=xyz)

, но это дало мне keyError"sum".

Единственный способ, которым это работает, - это получить sum и count как два списка, затем разбросайте эти два списка. Но это занимает много времени.

Есть ли какой-нибудь простой способ сразу получить эти два вложенных столбца?

Спасибо.

Answer 1

С вашей картинки похоже, что у вас есть MultiIndex для столбцов. В этом случае вы можете получить доступ к столбцам путем индексации с помощью кортежа, который задает значения для каждого уровня:

sns.scatterplot(df[('Total', 'sum')], df[('Total', 'count')])