将每个数字数组元素绘制为水平线

Question

有没有一种简单的方法可以将每个数组元素绘制为python中的水平线？

例:

some_array = [2 4 5 8 4 3 ... n]

现在我想要一个绘制水平线的图形:

y_1 = 2
y_2 = 4
y_3 = 5
.
.
.
y_n = n

some_array的长度不固定,因为我不知道它将包含多少元素,因此代码必须能够处理这个.

我知道我可以这样做:

plt.plot((x[0], x[-1]), (some_array[0], some_array[0]), 'k-')

并且可能把它放在while循环中,所以它运行数组的长度,但我觉得有更好的方法.

Answer 1

用途plt.hlines:

some_array = [2, 4, 5, 8, 4, 3]
plt.hlines(some_array, 0, 1, colors=['b', 'g', 'r', 'y', 'c', 'm'])
plt.ylim(1, 9)