Python numpy索引混合元组索引和范围索引
我有一个带有形状(m,n,3)的numpy数组,我想用形状(100,2)的另一个数组idx索引到前两列。所以我想要的是:Python numpy索引混合元组索引和范围索引,python,arrays,numpy,multidimensional-array,Python,Arrays,Numpy,Multidimensional Array,我有一个带有形状(m,n,3)的numpy数组,我想用形状(100,2)的另一个数组idx索引到前两列。所以我想要的是: np.array([a[x,y,:] for x,y in idx]) 最有效的方法是什么?试试a[idx[:,0],idx[:,1],:]。也就是说,使用idx的整列作为第一和第二个索引。谢谢,这很有效!如果你能给我一个答案,那么我可以接受。
np.array([a[x,y,:] for x,y in idx])
最有效的方法是什么?试试
a[idx[:,0],idx[:,1],:]
。也就是说,使用idx的整列作为第一和第二个索引。谢谢,这很有效!如果你能给我一个答案,那么我可以接受。