1. 程式人生 > 實用技巧 >numpy 建立陣列

numpy 建立陣列

numpy 建立陣列


1.從已有列表轉換為陣列

import numpy as np


a = [1, 2, 3, 4]
array = np.asarray(a) # np.array(a)
print(array, type(array))

結果:

[1 2 3 4] <class 'numpy.ndarray'>

程式碼裡的也可以是元組、numpy陣列......,還有一個注意點兒就是,有小夥伴會問有些人用的是np.array而不是np.asarray.其實這兩個最後創建出來的陣列是一樣的,但是 array裡引數給的就是ndarray陣列物件時就會複製一份陣列,而asarray則是原陣列的引用。

2.zeros(全為0的陣列),ones(全為1的陣列),full(指定元素填衝整個陣列)

array1 = np.zeros((3, 4))
array2 = np.ones((3, 4))
array3 = np.full((3, 4), 5)


print(array1)
print(array2)
print(array3)

結果:

[[0. 0. 0. 0.]
 [0. 0. 0. 0.]
 [0. 0. 0. 0.]]
[[1. 1. 1. 1.]
 [1. 1. 1. 1.]
 [1. 1. 1. 1.]]
[[5 5 5 5]
 [5 5 5 5]
 [5 5 5 5]]

引數是zeros、ones引數陣列的形狀(shape), (3, 4)代表3行4列,full引數在形狀的基礎上加了一個填充值,陣列的每個元素將以該值填充