numpy 随机抽取维度未知的数组的行(第一维)

q9rjltbz  于 11个月前  发布在  其他
关注(0)|答案(1)|浏览(75)

当维数已知时,任务是微不足道的。以一个2D数组为例:

a = np.random.randint(10, a=(5,2))
a[np.random.choice(a.shape[0]), :]

字符串
但是,在我的函数中,数组的维数是任意的。如何处理这种情况?

vatpfxk5

vatpfxk51#

使用第一维的大小来确定随机范围:

a[np.random.randint(0,a.shape[0],10)]

字符串
或者如果你喜欢,包括一个省略号

a[np.random.randint(0,a.shape[0],10),...]


默认情况下,1索引数组从行中选择。

相关问题