python - 以 numpy 行索引为标记的 Matplotlib 散点图

Question

我有一个 numpy 数组，我正在尝试使用 matplotlib 用散点图绘制它。

from matplotlib import pyplot as plt
from matplotlib import pylab as pl

pl.plot(matrix[:,0],matrix[:,1], 'ro')

这给了我类似的东西:

现在我想用与 numpy 数组中行的索引相对应的数字替换红点。我该怎么做？

谢谢!

Answer 1

您可以使用 plt.text 执行此操作:

from matplotlib import pyplot as plt

import numpy as np
N = 100
matrix = np.random.rand(N,2)

plt.plot(matrix[:,0],matrix[:,1], 'ro', alpha = 0.5)
for i in range(matrix.shape[0]):
    plt.text(matrix[i,0], matrix[i,1], str(i))

plt.show()

如果要替换红点，则设置alpha = 0.0: