随机采样具有未知维度的数组的行(第一维)
Randomly sample rows (first dimension) of an array with unknown dimension
当维度已知时,任务是微不足道的。取一个二维数组:
a = np.random.randint(10, a=(5,2))
a[np.random.choice(a.shape[0]), :]
但是,在我的函数中,数组的维数是任意的。这种情况怎么处理?
使用第一个维度的大小来确定随机范围:
a[np.random.randint(0,a.shape[0],10)]
或者如果您愿意,可以包含省略号
a[np.random.randint(0,a.shape[0],10),...]
1 个索引数组默认从行中选择。
当维度已知时,任务是微不足道的。取一个二维数组:
a = np.random.randint(10, a=(5,2))
a[np.random.choice(a.shape[0]), :]
但是,在我的函数中,数组的维数是任意的。这种情况怎么处理?
使用第一个维度的大小来确定随机范围:
a[np.random.randint(0,a.shape[0],10)]
或者如果您愿意,可以包含省略号
a[np.random.randint(0,a.shape[0],10),...]
1 个索引数组默认从行中选择。