如何使用matplotlib.pyplot绘制通过不同点的线的图表？

Question

我对Python还不熟悉。我从一个例子中借用了这段代码，并输入了我自己的数字：

import matplotlib.pyplot as plt

plt.plot(1, 9, 'rs', 2, 8, 'bs', 7, 3, 'g^', 9, 1, 'r^')
plt.title('Chart of 4 companies')
plt.axis([0, 12, 0, 12])
plt.grid(True)
plt.show()

这很好，它给我看了一张图表，上面有4个点。但是有没有可能显示从点[0,0]到这些线的线呢？我想强调一下余弦相似性，但我失败得很厉害

Answer 1

如果你想要一条从（0，0）到点（x，y）的直线，你可以通过

plt.plot([0,x],[0,y])

所以整个代码可能是这样的

import matplotlib.pyplot as plt

plt.plot(1, 9, 'rs', 2, 8, 'bs', 7, 3, 'g^', 9, 1, 'r^')
plt.plot([0,1], [0,9], 'r')
plt.plot([0,2], [0,8], 'b')
plt.plot([0,7], [0,3], 'g')
plt.plot([0,9], [0,1], 'r')
plt.title('Chart of 4 companies')
plt.axis([0, 12, 0, 12])
plt.grid(True)
plt.show()