Python Matplotlib: 如何使用柱状图绘制两个数组的比较

以下代码演示了如何使用 Python Matplotlib 库绘制两个数组的比较，并将它们分别用柱状图绘制在同一张图的上下两部分。

import matplotlib.pyplot as plt
import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([5, 4, 3, 2, 1])
x = np.arange(len(a))

fig, (ax1, ax2) = plt.subplots(2, 1, sharex=True)

ax1.bar(x, a)
ax1.set_ylabel('a')

ax2.bar(x, b)
ax2.set_ylabel('b')
ax2.set_xlabel('x')

plt.show()

代码解释

• 首先定义了两个数组a和b，以及x作为横坐标，用np.arange()生成了[0, 1, 2, 3, 4]。
• 然后创建了一个subplot，分成两行一列，共享横坐标。ax1和ax2分别代表上下两个图像。
• 对于ax1，用bar()方法画柱状图，横坐标是x，纵坐标是a。用set_ylabel()方法设置y轴标签为a。
• 对于ax2，同样用bar()方法画柱状图，横坐标是x，纵坐标是b。用set_ylabel()方法设置y轴标签为b，用set_xlabel()方法设置x轴标签为x。最后用plt.show()显示图像。

运行这段代码，就可以得到如下的图像：

通过这种方式，可以清晰地比较两个数组的值，并直观地展示它们之间的关系。