Come convertire un set python in un array numpy?

Question

Sto usando un'operazione set in python per eseguire una differenza simmetrica tra due array numpy. Il risultato, tuttavia, è un set e ho bisogno di riconvertirlo in una serie numpy per andare avanti. C'è un modo per fare questo? Ecco quello che ho provato:

a = numpy.array([1,2,3,4,5,6]) 
b = numpy.array([2,3,5]) 
c = set(a)^set(b) 

Il risultato è un insieme:

In [27]: c 
Out[27]: set([1, 4, 6]) 

Se converto a un array NumPy, si pone l'intero set nel primo elemento dell'array.

In [28]: numpy.array(c) 
Out[28]: array(set([1, 4, 6]), dtype=object) 

Che cosa ho bisogno, tuttavia, sarebbe questo:

array([1,4,6],dtype=int) 

potevo ciclo sugli elementi di convertire uno per uno, ma avrò 100.000 elementi e sperato in una funzione built-in per salvare il ciclo. Grazie!

Answer 1

Non convertire la matrice NumPy ad un insieme di eseguire esclusivo o. Utilizzare direttamente setxor1d.

>>> import numpy 
>>> a = numpy.array([1,2,3,4,5,6]) 
>>> b = numpy.array([2,3,5]) 
>>> numpy.setxor1d(a, b) 
array([1, 4, 6]) 

Answer 2

Do:

>>> numpy.array(list(c)) 
array([1, 4, 6]) 

E DTYPE è int (Int64 dalla mia parte.)

Answer 3

Prova questo.

numpy.array(list(c)) 

Conversione elencare prima di inizializzare matrice NumPy fisserebbe i singoli elementi a intero piuttosto che il primo elemento come oggetto.