使用python在具有特定值的二维数组中选择随机索引

Question

所以说我有一个像这样的 numpy 数组：

[[2 2 2 2 2 2 2 2 2 2 2 2 2 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 2 2 2 2 2 2 2 2 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 0 0 0 0 2 2 2 2 2 2 2 2 2 1 1 1 1 1 1 1 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 0 0 0 0 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 1 1 1 1 1 1 1 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 2 1 1 1 1 1 1 1 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 2 2 2 2 2 2 2 2 2 2 1 1 2 2 0 0 0 2 1 1 1 1 1 1 1 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 0 0 0 0 2 1 1 2 0 0 0 0 2 2 2 1 1 2 2 2 2 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 0 0 0 0 2 1 1 2 0 0 0 0 0 0 2 1 1 2 0 0 0 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 0 0 0 0 2 1 1 2 0 0 0 0 0 0 2 1 1 2 0 0 0 0]
 [2 2 2 2 2 1 1 2 2 2 2 2 2 2 0 0 0 0 0 0 0 0 2 1 1 2 0 0 0 0 0 0 2 1 1 2 0 0 0 0]
 [0 0 0 0 2 1 1 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 2 1 1 2 0 0 0 0 0 0 2 1 1 2 0 0 0 0]
 [0 0 0 0 2 1 1 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 2 1 1 2 0 0 2 2 2 2 2 1 1 2 2 0 0 0]
 [0 0 0 0 2 1 1 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 2 1 1 2 2 0 2 1 1 1 1 1 1 1 2 0 0 0]
 [2 2 2 2 2 1 1 2 2 2 2 2 2 2 2 2 2 2 0 0 0 0 2 1 1 1 2 0 2 1 1 1 1 1 1 1 2 0 0 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 2 1 1 1 2 0 2 1 1 1 1 1 1 1 2 0 0 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 2 1 1 1 2 0 2 1 1 1 1 1 1 1 2 0 0 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 2 1 1 1 2 0 2 2 2 2 2 2 2 2 2 0 0 0]
 [2 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 2 0 0 0 0 2 2 2 2 2 0 0 0 0 0 0 0 0 0 0 0 0 0]
 [2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0]]

我想在数组中随机选择一个位置为 1，我该怎么做？

Answer 1

使用np.argwhere 和random.choice：

import numpy as np
import random

a = np.random.choice([0, 1, 2], (20, 20))
i = random.choice(np.argwhere(a == 1))

Answer 2

您可以使用numpy.where 来识别值为 1 的单元格的坐标，然后从此输出中选择一个（或多个）随机索引：

# test array
a = np.random.randint(0, 3, size=(15, 10))

# get coordinates where a == 1
coords = np.c_[np.where(a==1)]

# pick N random cells
N = 1
out = coords[np.random.choice(coords.shape[0], N, replace=False)]

# check
x,y = out[0]
assert a[x,y] == 1

示例：(5, 3)

一个：

array([[0, 1, 0, 1, 1, 2, 0, 2, 0, 0],
       [0, 2, 1, 2, 2, 0, 1, 1, 1, 1],
       [0, 1, 0, 0, 1, 2, 0, 2, 0, 1],
       [1, 2, 0, 1, 1, 1, 0, 2, 0, 2],
       [2, 0, 2, 0, 0, 0, 1, 1, 2, 0],
       [0, 1, 0, 1, 2, 2, 0, 1, 1, 1],
       [1, 2, 2, 2, 0, 2, 1, 0, 1, 2],
       [0, 0, 2, 0, 0, 0, 0, 0, 0, 2],
       [0, 2, 1, 1, 1, 0, 1, 1, 1, 0],
       [1, 2, 0, 1, 2, 0, 2, 0, 1, 2],
       [2, 1, 0, 1, 1, 0, 2, 2, 2, 2],
       [1, 2, 2, 2, 2, 2, 0, 1, 2, 2],
       [1, 2, 1, 0, 2, 2, 0, 2, 0, 0],
       [2, 0, 2, 2, 2, 0, 0, 0, 1, 2],
       [0, 1, 2, 2, 2, 1, 0, 0, 0, 0]])