通过索引从Pytorch或Numpy 2D数组中快速删除多行的方法

您可以创建一个初始 mask (布尔)数组 True 对于要删除的元素,然后将其反转以给出 面具 的元素。

remove_mask = np.zeros(my_array.shape[0], dtype=bool)
remove_mask[remove_ixs] = True
mask = ~remove_mask
    
new_array = my_array[mask, :]

或者全部启动 真的 相反:

mask = np.ones(my_array.shape[0], dtype=bool)
mask[remove_ixs] = False
    
new_array = my_array[mask, :]

出于某种原因,对于较小的阵列,第一个版本的速度更快。

您可以使用 np.delete() :

import numpy as np

A = np.random.rand(5000000, 3)
remove_ixs = np.random.choice(5000000, 50000, replace=False)
B = np.delete(A, remove_ixs, axis=0)

print(len(B))

打印

4950000