Chuyển mảng NumPy sang cấu trúc Danh sách Python?

Question

Làm cách nào để chuyển đổi một mảng NumPy thành Danh sách Python (ví dụ: [[1,2,3],[4,5,6]]) và thực hiện nhanh chóng một cách hợp lý?

Answer 1

Sử dụng tolist():

import numpy as np 
>>> np.array([[1,2,3],[4,5,6]]).tolist() 
[[1, 2, 3], [4, 5, 6]] 

Lưu ý rằng điều này chuyển đổi các giá trị từ bất cứ loại NumPy họ có thể có (ví dụ np.int32 hoặc np.float32) để "loại Python tương thích gần nhất" (trong danh sách) . Nếu bạn muốn bảo toàn các kiểu dữ liệu khó khăn, bạn có thể gọi danh sách() trên mảng của bạn thay vào đó, và bạn sẽ kết thúc với một danh sách numpy scalars. (Nhờ Mr_and_Mrs_D đã chỉ ra rằng trong một chú thích.)

Answer 2

mảng NumPy có một tolist method:

In [1]: arr=np.array([[1,2,3],[4,5,6]]) 

In [2]: arr.tolist() 
Out[2]: [[1, 2, 3], [4, 5, 6]] 

Answer 3

Phương pháp ToList NumPy tạo mảng lồng nhau nếu hình dạng mảng NumPy là 2D.

nếu muốn danh sách phẳng, phương thức bên dưới hoạt động.

import numpy as np 
from itertools import chain 

a = [1,2,3,4,5,6,7,8,9] 
print type(a), len(a), a 
npa = np.asarray(a) 
print type(npa), npa.shape, "\n", npa 
npa = npa.reshape((3, 3)) 
print type(npa), npa.shape, "\n", npa 
a = list(chain.from_iterable(npa)) 
print type(a), len(a), a`