查找从与条件匹配的np.array元素中随机选择的一个元素的索引

时间:2019-08-18 21:23:06

标签: python numpy-ndarray

考虑下面的numpy数组arr1

arr1 = np.array([-8,-3,-5,-2, 8,-4],
                [ 7, 4, 3, 1,-5, 6],  
                [-6, 1,-2,-6,-4,-8],
                [ 5,-2, 7,-5,-3,-1]]) 

row_idx = 2
val = -6

我想从指定的行索引(即:2)中随机选择一个值(即:-6)的索引。我首先隔离目标值:

arr2 = np.where(arr1[row_idx][:] == val)
print(arr2)

这可以正确生成所需的索引:(array([0, 3)], dtype=int64,)

我是一个足够新的编码器,遇到了随机选择0或3 的麻烦。什么是有效的方法? (实际的arr1更大)。

谢谢

0 个答案:

没有答案