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

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：