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])

猜你喜欢

转载自blog.csdn.net/Ahead_J/article/details/85107196
今日推荐