Question

我试图创建条形图的子图。 matplotlib很有趣，它标记了条形图的x轴，所以你需要传递一个索引然后使用xticks函数来传递标签。我想创建2个子图，每个子图都有相同的x轴标签但是下面的代码我只能将标签传递到最后一个条形图。我的问题是如何将标签传递给两个条形图？

fig, axes = plt.subplots(nrows=2, ncols=1, figsize=(18,15))

r1 = axes[0].bar(idx, data1, align='center')
axes[0].set_title('title1')

r2 = axes[1].bar(idx, data2, align='center')
axes[1].set_title('title2')

plt.xticks(idx, idx_labels, rotation=45)

Answer 1

plt.xticks，因为所有其他pyplot命令与当前活动的轴相关。通过plt.subplots()一次生成多个轴，其中最后一个是当前轴，因此plt.xticks将设置刻度的轴。

您可以使用plt.sca(axes)设置当前轴：

import matplotlib.pyplot as plt

idx, data1, data2, idx_labels = [1,2,3], [3,4,2], [2,5,4], list("ABC")

fig, axes = plt.subplots(nrows=2, ncols=1)

r1 = axes[0].bar(idx, data1, align='center')
axes[0].set_title('title1')

plt.sca(axes[0])
plt.xticks(idx, idx_labels, rotation=45)

r2 = axes[1].bar(idx, data2, align='center')
axes[1].set_title('title2')

plt.sca(axes[1])
plt.xticks(idx, idx_labels, rotation=45)

plt.show()

但是，在使用子图时，使用matplotlib的面向对象API而不是pyplot statemachine通常更容易。在这种情况下，您使用ax.set_xticks和ax.set_xticklabels，其中ax是您要为其设置某些属性的轴。这更直观，因为从代码中可以很容易地看出它正在处理哪些轴。

import matplotlib.pyplot as plt

idx, data1, data2, idx_labels = [1,2,3], [3,4,2], [2,5,4], list("ABC")

fig, axes = plt.subplots(nrows=2, ncols=1)

r1 = axes[0].bar(idx, data1, align='center')
axes[0].set_title('title1')

r2 = axes[1].bar(idx, data2, align='center')
axes[1].set_title('title2')

for ax in axes:
    ax.set_xticks(idx)
    ax.set_xticklabels(idx_labels, rotation=45)

plt.show()

将相同的x轴标签传递给条形图的matplotlib子图

1 个答案: