Come posso trovare righe identiche in array diversi indipendentemente dall'ordine degli elementi usando Numpy? [duplicare]
Aug 19 2020
Stavo lavorando a un progetto e mi sono imbattuto in questo problema. Ho due array A e B di forma (8,3) e (2,2). Come posso trovare tutte le righe di A che contengono elementi di ogni riga di B indipendentemente dall'ordine degli elementi in B?
A = np.random.randint(0,5,(8,3))
B = np.random.randint(0,5,(2,2))
Grazie!
Risposte
2 Geom Aug 19 2020 at 06:05
Ecco un modo per farlo:
Import numpy as np
A = np.random.randint(0,5,(8,3))
B = np.random.randint(0,5,(2,2))
C = (A[..., np.newaxis, np.newaxis] == B)
rows = np.where(C.any((3,1)).all(1))[0]
print(rows)
Produzione:
[0 2 3 4]
2 BrycePaule Aug 19 2020 at 06:14
import numpy as np
A = np.random.randint(0, 5, (8, 3))
B = np.random.randint(0, 5, (2, 2))
for BRow in B:
for ARow in A:
if all(item in ARow for item in BRow):
print(f'{BRow} in {ARow}')
Tuttavia, questo non controlla i duplicati, ad esempio accetta [3, 3] in [1, 2, 3]