为什么绘制数据透视表时，图例的标题中没有'None'？

Posted 2021-05-06

我有一个数据框，然后调用pivot_table方法，然后绘制结果。

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame({'Date': ['2020-04-10', '2020-04-11', '2020-04-12', '2020-04-13','2020-04-10', '2020-04-11', '2020-04-12', '2020-04-13'],
'Station': ['Hogwarts', 'Hogwarts', 'Hogwarts', 'Hogwarts','Hogwarts', 'Hogwarts', 'Hogwarts', 'Hogwarts'],
'Direction': ['Southbound', 'Southbound', 'Southbound', 'Southbound','Northbound','Northbound','Northbound','Northbound'],
'Daily trains': [1,1,2,3,0,0,2,1]})

df['Date'] = pd.to_datetime(df['Date'],format="%Y-%m-%d")

df1 = df.pivot_table(index='Date', columns=['Station','Direction'],values=['Daily trains'])
fig, ax = plt.subplots(1)

df1.plot(ax=ax)
ax.figure.savefig('so.png',bbox_inches='tight')

这里是结果：

为什么我在图例的标题中得到'None'？

答案

它标记列名称。

df1.columns.names
FrozenList([None, 'Station', 'Direction'])

您提供的是list值，但只有一个值，因此它在您的MultiIndex列中创建了不必要的级别。删除那个。

...

df1 = df.pivot_table(index='Date', columns=['Station','Direction'], values='Daily trains')
fig, ax = plt.subplots(1)

df1.plot(ax=ax)
ax.figure.savefig('so.png',bbox_inches='tight')