numpy - ndarray - So entfernen Sie Zeilen basierend auf einem anderen Array

Question

Ich möchte Zeilen aus einem ndarray basierend auf einem anderen Array entfernen. zum Beispiel:

k = [1,3,99] 

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

erwartet Ergebnis:

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

die erste Spalte der Zeilen mit den Werten in k wird

Answer 1

Sie np.in1d zwischen einer Maske von Spielen erstellen können entfernt werden die erste Spalte von n und k und dann die invertierte Maske verwenden, um die nicht übereinstimmenden Zeilen aus n auszuwählen, so -

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

Wenn die erste Spalte bereits von int dtype ist, überspringen Sie den .astype(int) Konvertierungsschritt.

Probelauf -

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

Für peformance, wenn die erste Spalte sortiert wird, können wir verwenden np.searchsorted -

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

Answer 2

Wenn, einfache Lösung, wie unten Verzeichnen Sie bitte Ihre Datenstruktur finden Sie können jedoch in list by list() -Methode konvertieren.

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