矩阵中的样本条目满足给定的要求

Question

有一个0-1矩阵，我需要从该矩阵中抽取M个不同的1值的条目。有这种需求有效的Python实现吗？基准线方法是在每次迭代过程中进行M次迭代，如果它的值为1，则随机取样1，然后保留它并保存它的位置，否则继续此迭代直到找到值为1的入口;并继续下一次迭代。这似乎不是一个好的启发式。

Answer 1

我选择了间接指数从numpy.nonzero

通过ndx_ndx名单上使用pop（）方法来获得一个（间接）索引输入数组不用更换回

最终ndx_ndx会当你有被清空得到所有的那些

import numpy as np 


ary = np.random.randint(2, size=(20, 20)) 

# get the indices of all of the ones 

ndx_ary_ones = np.nonzero(ary) 

# make a range list for pointing into ndx_ary_ones 

ndx_ndx = list(range(len(ndx_ary_ones[0]))) 

# randomize the order 

np.random.shuffle(ndx_ndx) 

# pop the last ndx_ndx 

a_ran_ndx_ndx = ndx_ndx.pop() 

# get the index tuple for the one in ary that we removed from ndx_ndx 

a_ran_one_ndx = (ndx_ary_ones[0][a_ran_ndx_ndx], 
       ndx_ary_ones[1][a_ran_ndx_ndx]) 

# testing... 

print('ary', ary, '\n') 
print('ndx_ary_ones ', *ndx_ary_ones, sep = '\n') 
print('\n','ndx_ndx[0:10] ', ndx_ndx[0:10], '\n') 

for _ in range (10): 
    a_ran_ndx_ndx = ndx_ndx.pop() 
    a_ran_one_ndx = (ndx_ary_ones[0][a_ran_ndx_ndx], 
        ndx_ary_ones[1][a_ran_ndx_ndx]) 
    print(a_ran_one_ndx, ary[a_ran_one_ndx]) 

ary [[0 0 0 ..., 1 1 1] 
[0 1 1 ..., 1 1 1] 
[1 0 0 ..., 1 0 1] 
..., 
[1 1 0 ..., 1 0 1] 
[1 1 0 ..., 1 1 1] 
[1 0 0 ..., 0 0 1]] 

ndx_ary_ones 
[ 0 0 0 ..., 19 19 19] 
[ 3 5 7 ..., 14 15 19] 

ndx_ndx[0:10] [121, 43, 146, 69, 64, 3, 29, 186, 98, 30] 

(7, 12) 1 
(8, 18) 1 
(0, 3) 1 
(10, 2) 1 
(18, 18) 1 
(17, 7) 1 
(15, 14) 1 
(4, 11) 1 
(10, 1) 1 
(4, 4) 1 

Answer 2

我们能够做到以如下方式组成：第一获得矩阵A，其中A [X，Y] = 1的所有的（X，Y）的元组（索引）。有k个这样的指数。现在滚动一个K边无偏骰子M次（我们可以使用函数randint(1,k)从均匀分布中抽取样本来模拟）。如果你想要更换样品（矩阵的相同位置可以选择多次），那么可以用M个函数调用完成。否则，对于有替换的样品（不允许重复），您需要跟踪已选择的位置，并在下次投掷潜水之前从阵列中删除这些索引。