如何为多索引数据框创建绘图

Question

我需要为每个游戏平台制作绘图（y = 'total_sales_sum'、x = 'year_of_release'）。为此，我使用了 pivot table，因此得到了多索引数据框。

data_recent_decade=data.query('year_of_release>=2006').pivot_table(index=['platform','year_of_release'],values=['total_sales'], aggfunc=['sum'])
data_recent_decade.columns=['total_sales_sum']
data_recent_decade.info()
for platform in data_recent_decade:
    data_recent_decade.plot(y='total_sales_sum', marker='o',grid=True,figsize=(13,4))
    plt.title(platform)
    plt.show()

这是最终的数据帧：

这是data_recent_decade.info()

<class 'pandas.core.frame.DataFrame'> MultiIndex: 101 entries, (3DS, 2011.0) to (XOne, 2016.0) Data columns (total 1 columns): total_sales_sum 101 non-null float64 dtypes: float64(1) memory usage: 1.4+ KB

我的破剧情：

如何为各个平台制作剧情？

Answer 1

您可以循环使用 pandas.MultiIndex：

for date, new_df in df.groupby(level = 0)

完整代码

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame({'platform': ['3DS', '3DS', '3DS', '3DS', '3DS', 'XB', 'XBOne', 'XBOne', 'XBOne', 'XBOne'],
                   'year_of_release': [2011, 2012, 2013, 2014, 2015, 2008, 2013, 2014, 2015, 2016],
                   'total_sales_sum': [60.53, 51.01, 56.32, 43.07, 27.21, 0.18, 18.96, 54.07, 59.92, 25.82]})
df = df.set_index(['platform', 'year_of_release'])


fig, ax = plt.subplots()

for date, new_df in df.groupby(level = 0):
    ax.plot(new_df.index.get_level_values('year_of_release').values,
            new_df['total_sales_sum'],
            label = new_df.index.get_level_values('platform').values[0],
            marker = 'o',
            linestyle = '-')

ax.legend(frameon = True)

plt.show()

作为替代方案，您可以使用 seaborn.lineplot:

而无需任何循环

fig, ax = plt.subplots()

sns.lineplot(ax = ax,
             data = df,
             x = df.index.get_level_values('year_of_release'),
             y = df['total_sales_sum'],
             hue = df.index.get_level_values('platform'),
             marker = 'o')

plt.show()

如何为多索引数据框创建绘图

How to create a plot for a multiindex dataframe

python

plot

matplotlib

multi-index

pandas

完整代码