使用
advanced-indexing-
my_array[indexes, np.arange(indexes.shape[-1])]
如果使用索引列表建立索引
indexes以每列选择一个索引,请使用-
my_array[indexes, np.arange(len(indexes))]

使用
advanced-indexing-
my_array[indexes, np.arange(indexes.shape[-1])]
如果使用索引列表建立索引
indexes以每列选择一个索引,请使用-
my_array[indexes, np.arange(len(indexes))]