python - 如何在 2d numpy 数组中设置随机元素而不循环？

Question

我有一个由 0 组成的二维数组，我想将一些随机元素设置为 1。首先选择行，然后选择列，如下所示:

>>> A = np.zeros((100, 50))
>>> rows = np.random.choice(100, size = 10, replace = False)
>>> cols = np.random.randint(50, size = 10)
>>> A[rows][cols] = 1
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
IndexError: index 24 is out of bounds for axis 0 with size 10

当然我可以通过显式循环来解决问题:

>>> for row in rows: A[row][np.random.randint(50)] = 1

但我不想。我可以使用 numpy 来完成我想做的事情，而无需显式循环吗？

Answer 1

在一组括号中提供所有索引，如下所示
A[行数，列数] = 1