从2D阵列中的每一行中删除相同索引的最快方法

Question

我正在寻找最快的方法(最好是使用numpy(来删除2D数组每行中的索引列表。例如：

matrix = [[1,2,3,4,5],  
[4,5,6,7,8],
[7,8,9,10,11]]
indices_to_delete = [2,3]

现在的目标是从每一行中删除这些索引，以获得：

result = [[1,2,5],  
[4,5,8],
[7,8,11]]

我目前的方法是使用对每一行单独执行此操作

result = []
for row in array:
result.append(np.delete(row, indices_to_delete))

有没有更快的方法？

Answer 1

您可以沿着不同的轴使用.delete：

>>> np.delete(matrix, indices_to_delete, axis=1)
array([[ 1,  2,  5],
[ 4,  5,  8],
[ 7,  8, 11]])