從二維數組中的每一行刪除相同索引的最快方法

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]])