1. 程式人生 > >python: numpy -- zeros,ones,eye函式

python: numpy -- zeros,ones,eye函式

 

  • zeros():可以用來構造全0陣列

如果是一維的,則需要一個括號,裡面的數字代表陣列的長度,如zeros(3)表示建立1行3列全為0的陣列

如果是多維的,則要有兩個括號,裡面的維度用tuple來表示,如zeros((3,3))表示建立3行3列全為0的陣列

>>> zeros(3)
array([ 0.,  0.,  0.])
>>> zeros((3,3))
array([[ 0.,  0.,  0.],
       [ 0.,  0.,  0.],
       [ 0.,  0.,  0.]])
  • ones(): 可以用來構造全1的陣列

如果是多維的,則要有兩個括號,裡面的維度用tuple來表示,如a=ones((3,1))表示建立3行1列全為1的陣列

如果是一維的,則需要一個括號,裡面的數字代表陣列的長度,如a=ones(3)表示建立1行3列全為1的陣列

a=ones((3,1))
print("a:\n",a)
print("type(a):",type(a))
b=ones(3)
print("b:\n",b)
print("type(b):",type(b))


#執行結果如下:
a:
 [[1.]
 [1.]
 [1.]]
type(a): <class 'numpy.ndarray'>
b:
 [1. 1. 1.]
type(b): <class 'numpy.ndarray'>
>>> ones((3,3))
array([[ 1.,  1.,  1.],
       [ 1.,  1.,  1.],
       [ 1.,  1.,  1.]])

 

  • eye(): 建立一個正方的NxN單位矩陣(對角線為1,其餘為0)
>>> eye(3)
array([[ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  0.,  1.]])