numpy 建立陣列
阿新 • • 發佈:2020-07-27
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引數在形狀的基礎上加了一個填充值,陣列的每個元素將以該值填充