假设我有一组点
x
和一组相应的数据y
。我现在将它们绘制在散点图中,plt.scatter(x,y)
。我得到的图包含一些由 matplotlib 生成的 x 轴刻度。有没有一种方法可以实现自动勾选,但要添加从 x 轴到散点中的点的垂直线并标记它们?
pyplot.stem
的方法,更容易使用:
import matplotlib.pyplot as plt
import numpy as np
x, y = np.random.random((2, 20))
fig, ax = plt.subplots()
ax.stem(x, y, markerfmt=' ')
plt.show()
如果您希望在行首添加项目符号,只需删除
markerfmt
。
像这样吗?
如果是这样,那么这里是要点。
import matplotlib.pyplot as plt
import numpy as np
from matplotlib import collections as matcoll
x = np.arange(1,13)
y = [15,14,15,18,21,25,27,26,24,20,18,16]
lines = []
for i in range(len(x)):
pair=[(x[i],0), (x[i], y[i])]
lines.append(pair)
linecoll = matcoll.LineCollection(lines)
fig, ax = plt.subplots()
ax.add_collection(linecoll)
plt.scatter(x,y)
plt.xticks(x)
plt.ylim(0,30)
plt.show()
附录:
对于彩色点,将
plt.scatter(x,y)
替换为:
colours = ['Crimson', 'Blue', 'Fuchsia', 'Gold', 'Green', 'Tomato', 'Indigo', 'Turquoise', 'Brown', 'Wheat', 'Yellow',]
plt.scatter(x,y,c=colours)
附录2:
根据要求,最简单的方法可能是使用第二个答案中提供的建议,并进行适当调整。
import matplotlib.pyplot as plt
from datetime import datetime
from random import randint
x = [datetime(2019, 6, i) for i in range(1,21)]
y = [randint(10,20) for i in range(1,21)]
fig, ax = plt.subplots()
fig.subplots_adjust(bottom=0.2)
plt.xticks(rotation=90)
ax.stem(x, y, markerfmt=' ')
plt.show()