索引具有2个索引列表的2D Numpy数组

Question

索引具有2个索引列表的2D Numpy数组我有一个奇怪的情况。我有一个2D Numpy数组，x：x = np.random.random_integers(0,5,(20,8))我有2个索引器 - 一个带有行的索引，另一个带有索引。为了索引X，我必须执行以下操作：row_indices = [4,2,18,16,7,19,4]col_indices = [1,2]x_rows = x[row_indices,:]x_indexed = x_rows[:,column_indices]而不仅仅是：x_new = x[row_indices,column_indices]（失败的：错误，无法用（2，）广播（20，））我希望能够使用广播在一行中进行索引，因为这样可以保持代码的清晰和可读性......而且，我不知道所有关于python的内容，但据我所知它，它应该更快一行（我将使用相当大的数组）。测试用例：x = np.random.random_integers(0,5,(20,8))row_indices = [4,2,18,16,7,19,4]col_indices = [1,2]x_rows = x[row_indices,:]x_indexed = x_rows[:,col_indices]x_doesnt_work = x[row_indices,col_indices]

Answer 1

import numpy as np
x = np.random.random_integers(0,5,(4,4))x
array([[5, 3, 3, 2],
       [4, 3, 0, 0],
       [1, 4, 5, 3],
       [0, 4, 3, 4]])# This indexes the elements 1,1 and 2,2 and 3,3indexes = (np.array([1,2,3]),np.array([1,2,3]))x[indexes]# returns array([3, 5, 4])

请注意，numpy具有非常不同的规则，具体取决于您使用的索引类型。所以索引几个要素应该是由tuple的np.ndarray（见索引手册）。

因此，您只需要转换list为np.ndarray，它应该按预期工作。