my_array[indexes, np.arange(indexes.shape[-1])]
如果使用索引列表建立索引indexes
以每列选择一个索引,请使用-
my_array[indexes, np.arange(len(indexes))]
数组(具有行索引)的numpy数组的Pythonic方法
my_array[indexes, np.arange(indexes.shape[-1])]
如果使用索引列表建立索引indexes
以每列选择一个索引,请使用-
my_array[indexes, np.arange(len(indexes))]