cách hiệu quả để chuyển đổi dấu phân cách tách chuỗi để NumPy mảng

Question

Tôi có một String như sau:

1|234|4456|789 

tôi phải chuyển đổi nó thành array.I NumPy muốn biết way.Since hiệu quả nhất mà tôi sẽ gọi chức năng này trong hơn 50 triệu lần!

Answer 1

Cách nhanh nhất là sử dụng phương pháp numpy.fromstring:

>>> import numpy 
>>> data = "1|234|4456|789" 
>>> numpy.fromstring(data, dtype=int, sep="|") 
array([ 1, 234, 4456, 789]) 

Answer 2

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

import numpy as np 
s = '1|234|4456|789' 
array = np.array([int(x) for x in s.split('|')]) 

... Giả sử rằng những con số đều ints. nếu không, hãy thay thế int bằng float trong đoạn mã trên.

EDIT 1:

Ngoài ra, bạn có thể làm điều này, nó sẽ chỉ tạo một danh sách trung gian (một trong những tạo ra bởi split()):

array = np.array(s.split('|'), dtype=int) 

EDIT 2:

Và cách khác, có thể nhanh hơn (cảm ơn tất cả nhận xét, guys!):

array = np.fromiter(s.split("|"), dtype=int) 

Answer 3

@jterrace thắng một (1) internet.

Trong các phép đo bên dưới mã ví dụ đã được rút ngắn để cho phép các phép thử vừa với một dòng mà không cần cuộn khi có thể.

Đối với những người không quen thuộc với timeitthe -s flag allows you to specify a bit of code which will only be executed once.

---

Cách nhanh nhất và ít nhất-lộn xộn là sử dụng numpy.fromstring như jterrace gợi ý:

python -mtimeit -s"import numpy;s='1|2'" "numpy.fromstring(s,dtype=int,sep='|')" 
100000 loops, best of 3: 1.85 usec per loop 

---

Ba ví dụ sau đây sử dụng string.split kết hợp với công cụ khác.

string.split với numpy.fromiter

python -mtimeit -s"import numpy;s='1|2'" "numpy.fromiter(s.split('|'),dtype=int)" 
100000 loops, best of 3: 2.24 usec per loop 

string.split với int() cast qua máy phát-biểu

python -mtimeit -s"import numpy;s='1|2'" "numpy.array(int(x) for x in s.split('|'))" 
100000 loops, best of 3: 3.12 usec per loop 

string.split với NumPy mảng kiểu int

python -mtimeit -s"import numpy;s='1|2'" "numpy.array(s.split('|'),dtype=int)" 
100000 loops, best of 3: 9.22 usec per loop