我正在尝试绘制时间段。我有两个'开始''结束'点的ndarrys。 我想把它作为一个数字上的块。请记住,块不是连续的,这正是我正在寻找的差距。 到现在为止我尝试使用补丁:
for x_1 , x_2 in zip(s_data['begin'].values ,s_data['end'].values):
ax1.add_patch(Rectangle((x_1,0),x_2-x_1,0.5))
plt.show()
但它只给我一个蓝色的数字。
虽然我想要这样的东西
答案 0 :(得分:5)
方法是正确的。您只需缩放轴,使完整的绘图在其范围内。
import matplotlib.pyplot as plt
import pandas as pd
df = pd.DataFrame({"begin": [1,4,6,9], "end" : [3,5,8,12]})
fig, ax = plt.subplots()
for x_1 , x_2 in zip(df['begin'].values ,df['end'].values):
ax.add_patch(plt.Rectangle((x_1,0),x_2-x_1,0.5))
ax.autoscale()
ax.set_ylim(-2,2)
plt.show()
值得注意的是matplotlib有一个函数broken_barh
,它简化了这些图表的创建。
import matplotlib.pyplot as plt
import pandas as pd
df = pd.DataFrame({"begin": [1,4,6,9], "end" : [3,5,8,12]})
fig, ax = plt.subplots()
ax.broken_barh(list(zip(df["begin"].values, (df["end"] - df["begin"]).values)), (0, 0.5))
ax.set_ylim(-2,2)
plt.show()
给出与上面相同的图表。