13

我在 Ubuntu 10.0.4 上使用 matplotlib 1.2.x 和 Python 2.6.5。我正在尝试创建一个由顶部图和底部图组成的单一图。

X 轴是时间序列的日期。顶部图包含数据的烛台图，底部图应包含条形图 - 具有自己的 Y 轴（也在左侧 - 与顶部图相同）。这两个图不应重叠。

这是我到目前为止所做的一个片段。

datafile = r'/var/tmp/trz12.csv'
r = mlab.csv2rec(datafile, delimiter=',', names=('dt', 'op', 'hi', 'lo', 'cl', 'vol', 'oi'))

mask = (r["dt"] >= datetime.date(startdate)) & (r["dt"] <= datetime.date(enddate))
selected = r[mask]
plotdata = zip(date2num(selected['dt']), selected['op'], selected['cl'], selected['hi'], selected['lo'], selected['vol'], selected['oi'])

# Setup charting 
mondays = WeekdayLocator(MONDAY)        # major ticks on the mondays
alldays    = DayLocator()               # minor ticks on the days
weekFormatter = DateFormatter('%b %d')  # Eg, Jan 12
dayFormatter = DateFormatter('%d')      # Eg, 12
monthFormatter = DateFormatter('%b %y')

# every Nth month
months = MonthLocator(range(1,13), bymonthday=1, interval=1)

fig = pylab.figure()
fig.subplots_adjust(bottom=0.1)
ax = fig.add_subplot(111)
ax.xaxis.set_major_locator(months)#mondays
ax.xaxis.set_major_formatter(monthFormatter) #weekFormatter
ax.format_xdata = mdates.DateFormatter('%Y-%m-%d')
ax.format_ydata = price
ax.grid(True)

candlestick(ax, plotdata, width=0.5, colorup='g', colordown='r', alpha=0.85)

ax.xaxis_date()
ax.autoscale_view()
pylab.setp( pylab.gca().get_xticklabels(), rotation=45, horizontalalignment='right')

# Add volume data 
# Note: the code below OVERWRITES the bottom part of the first plot
# it should be plotted UNDERNEATH the first plot - but somehow, that's not happening
fig.subplots_adjust(hspace=0.15)
ay = fig.add_subplot(212)
volumes = [ x[-2] for x in plotdata]
ay.bar(range(len(plotdata)), volumes, 0.05)

pylab.show()

我已经设法使用上面的代码显示了这两个图，但是，底部图有两个问题：

  1. 它完全覆盖了第一个（顶部）图的底部 - 几乎就像第二个图在与第一个图相同的“画布”上绘制 - 我看不出在哪里/为什么会发生这种情况。

  2. 它用自己的索引覆盖现有的 X 轴，X 轴值（日期）应该在两个图之间共享。

我在代码中做错了什么？有人能发现是什么导致第二个（底部）图覆盖了第一个（顶部）图 - 我该如何解决这个问题？

这是上面代码创建的绘图的屏幕截图：

错误的情节

[[编辑]]

按照 hwlau 的建议修改代码后，这是新的情节。它比第一个更好，因为这两个地块是分开的，但是仍然存在以下问题：

  1. X 轴应由两个图共享（即 X 轴应仅显示为第二个 [底部] 图）

  2. 第二个图的 Y 值似乎格式不正确

部分正确的情节

我认为这些问题应该很容易解决，但是我的 matplotlib fu 目前不是很好，因为我最近才开始使用 matplotlib 编程。任何帮助都感激不尽。

4

3 回答 3

12

您的代码似乎有几个问题：

  1. 如果您使用figure.add_subplots它的完整签名subplot(nrows, ncols, plotNum)可能会更明显，您的第一个图要求 1 行和 1 列，而第二个图要求 2 行和 1 列。因此，您的第一个情节正在填充整个数字。而不是fig.add_subplot(111)后面跟着 fig.add_subplot(212) usefig.add_subplot(211)后面跟着fig.add_subplot(212).

  2. 共享轴应该在add_subplot命令中使用sharex=first_axis_instance

我整理了一个您应该能够运行的示例：

import matplotlib.pyplot as plt
import matplotlib.ticker as mticker
import matplotlib.dates as mdates


import datetime as dt


n_pts = 10
dates = [dt.datetime.now() + dt.timedelta(days=i) for i in range(n_pts)]

ax1 = plt.subplot(2, 1, 1)
ax1.plot(dates, range(10))

ax2 = plt.subplot(2, 1, 2, sharex=ax1)
ax2.bar(dates, range(10, 20))

# Now format the x axis. This *MUST* be done after all sharex commands are run.

# put no more than 10 ticks on the date axis.  
ax1.xaxis.set_major_locator(mticker.MaxNLocator(10))
# format the date in our own way.
ax1.xaxis.set_major_formatter(mdates.DateFormatter('%Y-%m-%d'))

# rotate the labels on both date axes
for label in ax1.xaxis.get_ticklabels():
    label.set_rotation(30)
for label in ax2.xaxis.get_ticklabels():
    label.set_rotation(30)

# tweak the subplot spacing to fit the rotated labels correctly
plt.subplots_adjust(hspace=0.35, bottom=0.125)

plt.show()

希望有帮助。

于 2012-04-04T12:27:12.697 回答
6

你应该改变这一行：

ax = fig.add_subplot(111)

到

ax = fig.add_subplot(211)

原始命令意味着有一行和一列，因此它占据了整个图形。所以你的第二张图 fig.add_subplot(212) 覆盖了第一张图的下部。

编辑

如果您不想要两个图之间的间隙，请使用 subplots_adjust() 更改子图边距的大小。

于 2012-04-04T10:18:56.843 回答
2

@Pelson 的示例已简化。

import matplotlib.pyplot as plt
import datetime as dt

#Two subplots that share one x axis
fig,ax=plt.subplots(2,sharex=True)

#plot data
n_pts = 10
dates = [dt.datetime.now() + dt.timedelta(days=i) for i in range(n_pts)]
ax[0].bar(dates, range(10, 20))
ax[1].plot(dates, range(10))

#rotate and format the dates on the x axis
fig.autofmt_xdate()

共享 x 轴的子图在一行中创建，当您需要两个以上的子图时，这很方便：

fig, ax = plt.subplots(number_of_subplots, sharex=True)

要在 x 轴上正确格式化日期，我们可以简单地使用fig.autofmt_xdate()

共享 x x 轴

有关其他信息，请参阅pylab 示例中的共享轴演示和日期演示。此示例在 Python3、matplotlib 1.5.1 上运行

于 2016-01-28T12:59:47.590 回答