python - 使用 matplotlib 绘制两行标签棒

Question

我想有一个使用 matplotlib.pyplot 的情节，其中 xticks 排列成两行几个月和几年，如下图所示。我做了那个情节，只是使用dataframe.plot()，即最简单的熊猫情节。

当我使用此代码进行绘图时（因为我需要添加另一个子图，这就是不使用的原因dataframe.plot()），如何获得 xticks 标签的之前设置？

import matplotlib.pyplot as plt
figure, ax = plt.subplots()
ax.plot(xdata, ydata)

我得到这个情节的 xticks 标签

我尝试使用matplotlib.dates.DateFormatter，matplotlib.ticker但找不到正确的设置

Answer 1

您可以使用主要和次要定位器接近您想要的内容，DateFormatter如下所示：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib.dates

dr= pd.date_range("2014-01-01", "2017-06-30", freq="D")
df = pd.DataFrame({"dates":dr, "num":np.cumsum(np.random.randn(len(dr)))})
df["dates"] = pd.to_datetime(df["dates"])

fig, ax = plt.subplots()
ax.plot(df.dates, df.num)

ax.xaxis.set_minor_locator(matplotlib.dates.MonthLocator())
ax.xaxis.set_major_locator(matplotlib.dates.MonthLocator([1,7]))
ax.xaxis.set_major_formatter(matplotlib.dates.DateFormatter("%b\n%Y"))
plt.show()

要仅显示一月份的年份，而不显示其他月份的年份，您可能需要将DateFormatter

class MyMonthFormatter(matplotlib.dates.DateFormatter):
    def __init__(self, fmt="%b\n%Y", fmt2="%b", major=[1], tz=None):
        self.fmt2 = fmt2
        self.major=major
        matplotlib.dates.DateFormatter.__init__(self, fmt, tz=tz)
    def __call__(self, x, pos=0):
        if x == 0: raise ValueError('Error')
        dt = matplotlib.dates.num2date(x, self.tz)
        if dt.month in self.major: 
            return self.strftime(dt, self.fmt)
        else:
            return self.strftime(dt, self.fmt2)

ax.xaxis.set_minor_locator(matplotlib.dates.MonthLocator())
ax.xaxis.set_major_locator(matplotlib.dates.MonthLocator([1,7]))
ax.xaxis.set_major_formatter(MyMonthFormatter())
plt.show()