python - 检索 numpy 数组中具有特定值的随机项的索引

Question

我有一个二维 numpy 数组，其中包含小范围的不同值。我想检索一个随机索引，但具有特定值。例如说这是数组:

arr = [[1,2,1,1,3],
       [4,3,4,1,1],
       [2,3,1,2,1],
       [1,2,1,1,2],
       [3,1,3,4,2]]

我想要一个任意值为 1 的随机索引，例如 (2,2)。到目前为止，我能够做到这一点的唯一方法是采用任何随机索引，然后检查它是否包含我想要的值，如果不包含则重新绘制。这对于上面的示例来说没问题，但我需要从大型数组中检索许多随机索引。

我描述的过程是否有方法或算法？

它的应用是提供分类图像的随机像素来执行精度评估。

Answer 1

您将列表转换为 numpy.array，然后从 np.argwhere 结果中随机选择一个坐标:

import numpy as np
import random
random.choice(np.argwhere(np.array(arr)==value))