Làm thế nào để chuyển đổi một bộ python thành một mảng numpy?

Question

Tôi đang sử dụng một thao tác được đặt trong python để thực hiện sự khác biệt đối xứng giữa hai mảng numpy. Kết quả, tuy nhiên, là một bộ và tôi cần phải chuyển đổi nó trở lại một mảng numpy để di chuyển về phía trước. Có cách nào để làm việc này không? Đây là những gì tôi đã cố gắng:

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

Kết quả là một tập hợp:

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

Nếu tôi chuyển sang một mảng NumPy, nó đặt toàn bộ nhóm bằng các phần tử mảng đầu tiên.

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

gì tôi cần, tuy nhiên, sẽ là thế này:

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

tôi có thể lặp trên các yếu tố để chuyển đổi từng người một, nhưng tôi sẽ có 100.000 yếu tố và hy vọng cho một hàm built-in để lưu vòng lặp. Cảm ơn!

Answer 1

Không chuyển đổi mảng NumPy để một bộ để thực hiện độc quyền-hoặc. Sử dụng trực tiếp 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

Đỗ:

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

Và dtype là int (Int64 đứng về phía tôi.)

Answer 3

Hãy thử điều này.

numpy.array(list(c)) 

Chuyển đổi thành danh sách trước khi khởi tạo mảng khối sẽ đặt các phần tử riêng thành số nguyên chứ không phải phần tử đầu tiên làm đối tượng.