随机抽样具有未知维度的数组的行（第一维）

Question

当维度已知时，任务很简单。拿2D阵列：

  

a = np.random.randint（10，a =（5,2））

     

a [np.random.choice（a.shape [0]），：]

但是，在我的函数中，数组的维度是任意的。如何处理这种情况？

Answer 1

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

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

或者如果您愿意，请包含省略号

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

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