gpt4 book ai didi

python - 在 matplotlib 中绘制预测点和 ground_truth 点之间的线

转载 作者:太空宇宙 更新时间:2023-11-04 08:00:46 24 4
gpt4 key购买 nike

我有两个数据框,ground_truth 和 prediction(都是 pandas 系列)。最后,我想绘制所有预测点 和所有ground_truth 点,就像我已经做的那样。我想做的是在每个预测点和 ground_truth 点之间绘制一条线。这样这条线就是预测点 x1,y1 和 ground_truth 点 x2,y2 之间的连接。为了更好地理解,我附上了一张图片。黑线(通过油漆创建)是我想要做的。 example

这是我已经拥有的:

fig, ax = plt.subplots()

ax.plot(pred,'ro', label='Prediction', color = 'g')
ax.plot(GT,'^', label='Ground Truth', color = 'r' )

plt.xlabel('a')
plt.ylabel('b')
plt.title('test')

plt.xticks(np.arange(-1, 100, 5))
plt.style.use('ggplot')
plt.legend()
plt.show()

最佳答案

我想最简单和最容易理解的解决方案是在循环中绘制 predGT 之间的相应线条。

import matplotlib.pyplot as plt
import numpy as np
plt.rcParams['legend.numpoints'] = 1

#generate some random data
pred = np.random.rand(10)*70
GT = pred+(np.random.randint(8,40,size= len(pred))*2.*(np.random.randint(2,size=len(pred))-.5 ))

fig, ax = plt.subplots(figsize=(6,4))

# plot a black line between the
# ith prediction and the ith ground truth
for i in range(len(pred)):
ax.plot([i,i],[pred[i], GT[i]], c="k", linewidth=0.5)
ax.plot(pred,'o', label='Prediction', color = 'g')
ax.plot(GT,'^', label='Ground Truth', color = 'r' )

ax.set_xlim((-1,10))
plt.xlabel('a')
plt.ylabel('b')
plt.title('test')

plt.legend()
plt.show()

enter image description here

关于python - 在 matplotlib 中绘制预测点和 ground_truth 点之间的线,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40742241/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com