使用plot_date时，将强制刻度显示在特定日期

Question

此代码将产生下图：

import math
import matplotlib.pyplot as plt
from matplotlib.dates import (YEARLY,HOURLY, DateFormatter,
                              drange)
import datetime


date1 = datetime.datetime(1952, 1, 1, 1, 1, 1)
date2 = datetime.datetime(1952, 1, 1, 23, 59, 59)
delta = datetime.timedelta(minutes= 10)

dates = drange(date1, date2, delta)
y = [math.sin(x/10.0) for x,_ in enumerate(dates)]

fig, ax = plt.subplots()

plt.plot_date(dates, y)

plt.plot_date([dates[15],dates[121]], [y[15], y[121]], marker="*", c="yellow", markersize=20)

ax.xaxis.set_major_formatter(DateFormatter('%H:%M:%S'))
ax.xaxis.set_tick_params(rotation=45, labelsize=10)

fig.autofmt_xdate()

plt.show()

我希望指出何时有两个以星号标记的事件发生。

如何强制pyplot在两颗星的位置下方绘制刻度线？

我发现我需要使用自定义定位器。但是到底如何呢？看起来它只会产生均匀分布的标尺。

rule = rrulewrapper(???)  
loc = RRuleLocator(rule)

Answer 1

如果您已确定刻度线的位置，则可以使用FixedLocator。

import math
import matplotlib.pyplot as plt
from matplotlib.dates import DateFormatter, drange
from matplotlib.ticker import FixedLocator
import datetime


date1 = datetime.datetime(1952, 1, 1, 1, 1, 1)
date2 = datetime.datetime(1952, 1, 1, 23, 59, 59)
delta = datetime.timedelta(minutes= 10)

dates = drange(date1, date2, delta)
y = [math.sin(x/10.0) for x,_ in enumerate(dates)]

fig, ax = plt.subplots()

plt.plot_date(dates, y)

events, = plt.plot_date([dates[15],dates[121]], [y[15], y[121]], 
                   marker="*", c="r", markersize=20, ls="")

ax.xaxis.set_major_formatter(DateFormatter('%H:%M:%S'))
ax.xaxis.set_tick_params(which="major", rotation=30, labelsize=9)

ax.xaxis.set_minor_locator(FixedLocator(events.get_xdata()))
ax.xaxis.set_minor_formatter(DateFormatter('%H:%M:%S'))
ax.xaxis.set_tick_params(which="minor", rotation=30, labelsize=10, labelcolor="r")

plt.setp(ax.get_xticklabels(which="both"), ha="right")
plt.show()

我将特殊的刻度标签设为红色，因为它们与现有的刻度标签重叠。我想您需要自己决定如何最终显示它。