numpy - ndarray - 如何删除基于另一个阵列的行

Question

我想从基于另一个阵列的ndarray中删除行。 例如：

k = [1,3,99] 

n = [ 
    [1,'a'] 
    [2,'b'] 
    [3,'c'] 
    [4,'c'] 
    [.....] 
    [99, 'a'] 
    [100,'e'] 
] 

预期结果：

out = [ 
    [2,'b'] 
    [4,'c'] 
    [.....] 
    [100,'e'] 
] 

行与k中的值的第一列将被删除

Answer 1

可以使用np.in1d创建之间的匹配的掩模的n和k，然后将第一列中使用反转掩模来选择非匹配的行断n，像这样 -

n[~np.in1d(n[:,0].astype(int), k)] 

如果第一列已经是int dtype，则跳过.astype(int)转换步骤。

采样运行 -

In [41]: n 
Out[41]: 
array([['1', 'a'], 
     ['2', 'b'], 
     ['3', 'c'], 
     ['4', 'c'], 
     ['99', 'a'], 
     ['100', 'e']], 
     dtype='|S21') 

In [42]: k 
Out[42]: [1, 3, 99] 

In [43]: n[~np.in1d(n[:,0].astype(int), k)] 
Out[43]: 
array([['2', 'b'], 
     ['4', 'c'], 
     ['100', 'e']], 
     dtype='|S21') 

对于peformance，如果第一列进行排序，我们可以使用np.searchsorted -

mask = np.ones(n.shape[0],dtype=bool) 
mask[np.searchsorted(n[:,0], k)] = 0 
out = n[mask] 

Answer 2

如果你的数据结构列表，请在下面找到简单的解决方案，但是你可以通过list（）方法转换成列表。

def check(list): 
k=[1,3,99] 
if(list[0] not in k): 
    return list 

final_list = map(check,n) 
final_list = final_list.remove(None) 
print final_list