使用 Python 为 sns 中的 relplot 设置 x-ticks 的数量

Question

我们如何在 Python 中为 relplot 设置 x-ticks 的数量？我们的图表如下所示：

我们目前的图表包含 7 个月份中的日期，但我们希望将所有月份都包含为 x-ticks。我们想将 x-ticks 的数量设置为 12，并将以下内容作为 x 轴标签：

months = ['2020-01', '2020-02', '2020-03', '2020-04', '2020-05', '2020-06', '2020-07', '2020-08', '2020-09', '2020-10', '2020-11', '2020-12']

以下是生成图表的函数：

def create_lineplot(dataframe):

    months = mdates.MonthLocator()  # every month
    years_fmt = mdates.DateFormatter('%Y-%m')  # This is a format. Will be clear in Screenshot

    # Filtering data to only select relevant columns and data from the year 2020
    dataframe = dataframe[['dev_id', 'outside_temperature', 'datetime']]
    dataframe["datetime"] = pd.to_datetime(dataframe["datetime"])
    dataframe_2020 = dataframe[dataframe['datetime'].dt.year == 2020]

    fig, axes = plt.subplots(figsize=(20, 2))

    mdf = pd.melt(dataframe_2020, id_vars=['datetime', 'dev_id'], var_name=['Temperature'])

    g = sns.relplot(data=mdf, x='datetime', y='value', kind='line', hue='Temperature', height=5, aspect=3)
    g._legend.remove()

    axes.xaxis.set_major_locator(months)
    axes.xaxis.set_major_formatter(years_fmt)
    axes.xaxis.set_minor_locator(months)

    plt.xticks(rotation='vertical')
    plt.tight_layout()
    plt.legend(loc='upper left')

    plt.savefig('lineplot.png')

    plt.show()

我们的数据示例：

Answer 1

代码的主要问题是在使用figure-level function 时创建了一个虚拟无花果和轴。因此，sns.relplot 创建了自己的图形和轴。您可以使用axes = g.axes.flat[0] 抓住它的第一把斧头。使用图形级函数时，应删除对fig, axes = plt.subplots(...) 的调用。

或者，您可以调用相关的轴级函数并传递 ax：sns.lineplot(..., ax=axes)。

import matplotlib.pyplot as plt
from matplotlib import dates as mdates
import seaborn as sns
import pandas as pd
import numpy as np

dataframe = pd.DataFrame({'datetime': pd.date_range('20200101', periods=366, freq='D'),
                          'val1': np.random.normal(0.01, .2, 366).cumsum() + 20,
                          'val2': np.random.normal(0.03, .25, 366).cumsum() + 25})

months_locator = mdates.MonthLocator()  # every month
years_fmt = mdates.DateFormatter('%Y-%m')  # This is a format. Will be clear in Screenshot

dataframe["datetime"] = pd.to_datetime(dataframe["datetime"])
dataframe_2020 = dataframe[dataframe['datetime'].dt.year == 2020]

mdf = dataframe_2020.melt(id_vars=['datetime'], value_vars=['val1', 'val2'])

g = sns.relplot(data=mdf, x='datetime', y='value', hue='variable', kind='line', height=5, aspect=3)
g._legend.remove()
axes = g.axes.flat[0]

axes.xaxis.set_major_locator(months_locator)
axes.xaxis.set_major_formatter(years_fmt)

axes.legend(loc='upper left')
axes.tick_params(axis='x', rotation=90)
axes.margins(x=0)

plt.tight_layout()
plt.show()