在Matplotlib中自定义Y轴比例

Question

我在Candlestick OHLC图表上绘制了一些水平线；但是，我的目标是使图表在Y轴上显示每条线的值。 这是我的代码：

plt.figure( 1 )
plt.title( 'Weekly charts' )
ax = plt.gca()

minor_locator_w = allweeks
major_formatter_w = yearFormatter

ax.xaxis.set_minor_locator( minor_locator_w )
ax.xaxis.set_major_formatter( major_formatter_w )

candlestick_ohlc(ax, zip(df_ohlc_w[ 'Date2' ].map( mdates.datestr2num ),
                 df_ohlc_w['Open'], df_ohlc_w['High' ],
                 df_ohlc_w['Low'], df_ohlc_w['Close']), width=0.6, colorup= 'g' )

ax.xaxis_date()

ax.autoscale_view()

plt.setp(ax.get_xticklabels(), horizontalalignment='right')

historical_prices_200 = [ 21.53, 22.09, 22.31, 22.67 ]

horizontal_lines = historical_prices_200

x1 = Date2[ 0 ]
x2 = Date2[ len( Date2 ) - 1 ]

plt.hlines(horizontal_lines, x1, x2, color='r', linestyle='-')

plt.show()

这是我得到的输出：

是否可以在Y轴上显示所有价格？

Answer 1

您可以使用Axes类的get_yticks函数来获取图表当前刻度位置的列表，追加要显示其他刻度的位置，然后使用set_yticks功能来更新图表。

ax.hlines(horizontal_lines, x1, x2, color="r")
ax.set_yticks(np.append(ax.get_yticks(), horizontal_lines))

要更改刻度标签的颜色以匹配行，请执行以下操作：

plt.setp(ax.get_yticklabels()[-len(horizontal_lines):], color="r")

或者，如果轴开始变得有点混乱，则可以使用text function在右端（或适合的任何地方）标记线条：

ax.hlines(horizontal_lines, x1, x2, color="r")
for v in horizontal_lines:
    ax.text(x2, v, v, ha="left", va="center", color="r")

您可能需要调整x轴的限制以适合标签。