带折线图的条形图 - 使用非数字索引

时间:2017-10-09 10:00:39

标签: python python-3.x pandas matplotlib charts

我想在同一个图表上显示数据框的条形图,以及表示总和的折线图。 我可以为索引为数字或文本的框架执行此操作。但它并不适用于日期时间索引。 这是我使用的代码:

import datetime as dt
np.random.seed(1234)
data = np.random.randn(10, 2)
date = dt.datetime.today()
index_nums =  range(10)
index_text = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'k']
index_date = pd.date_range(date + dt.timedelta(days=-9), date)
a_nums = pd.DataFrame(columns=['a', 'b'], index=index_nums, data=data)   
a_text = pd.DataFrame(columns=['a', 'b'], index=index_text, data=data)
a_date = pd.DataFrame(columns=['a', 'b'], index=index_date, data=data)

fig, ax = plt.subplots(3, 1)
ax = ax.ravel()
for i, a in enumerate([a_nums, a_text, a_date]):
    a.plot.bar(stacked=True, ax=ax[i])
    (a.sum(axis=1)).plot(c='k', ax=ax[i])

enter image description here

正如您所看到的,最后一张图表仅作为带有条形图图例的行。并且缺少日期。

如果我用

替换最后一行
ax[i].plot(a.sum(axis=1), c='k')

然后:

  • index_nums的图表是相同的
  • 带有index_text的图表会引发错误
  • 带有index_date的图表显示条形图,但不显示折线图。

fgo我使用的是pytho 3.6.2 pandas 0.20.3和matplotlib 2.0.2

1 个答案:

答案 0 :(得分:2)

将条形图和线图绘制到相同的轴通常可能会有问题,因为条形图将条形图放在整数位置(0,1,2,...N-1),而线条图使用数字数据来确定纵坐标。

在问题的情况下,使用range(10)作为条形图和线条图的索引工作正常,因为这些正是条形图将使用的数字。使用文本也可以正常工作,因为这需要用数字代替才能显示它,当然前N个整数也用于此。

日期时间索引的条形图也使用前N个整数,而线条图将在日期上绘制。因此,根据哪一个首先出现,您只能看到线条或条形图(您实际上会通过相应地更改xlimits来看到另一个)。

一个简单的解决方案是首先绘制条形图并将索引重置为线图的数据框上的数字。

import matplotlib.pyplot as plt
import pandas as pd
import numpy as np; np.random.seed(1234)
import datetime as dt

data = np.random.randn(10, 2)
date = dt.datetime.today()

index_date = pd.date_range(date + dt.timedelta(days=-9), date)
df = pd.DataFrame(columns=['a', 'b'], index=index_date, data=data)

fig, ax = plt.subplots(1, 1)

df.plot.bar(stacked=True, ax=ax)
df.sum(axis=1).reset_index().plot(ax=ax)

fig.autofmt_xdate()  
plt.show()

或者,您可以像往常一样绘制线图,并使用matplotlib条形图,它接受数字位置。请参阅此答案:Python making combined bar and line plot with secondary y-axis