1. 程式人生 > >numpy的array與python標準庫中的list相互轉換

numpy的array與python標準庫中的list相互轉換

將list轉換成array,用np.array(<list>)

將array轉換為list,用<array>.tolist()

>>> a
array([[1., 2., 3.],
       [1., 2., 3.],
       [0., 0., 0.],
       [0., 0., 0.]])
# 這會將array轉換成每個元素都是array的list
>>> list(a)
[array([1., 2., 3.]), array([1., 2., 3.]), array([0., 0., 0.]), array([0., 0., 0.])]
# array -> list
>>> a.tolist()
[[1.0, 2.0, 3.0], [1.0, 2.0, 3.0], [0.0, 0.0, 0.0], [0.0, 0.0, 0.0]]
# list -> array
>>> np.array([1,2,3])
array([1, 2, 3])