考虑下面的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更大)。
谢谢