1. 程式人生 > >Python3 NumPy陣列生成的簡單命令

Python3 NumPy陣列生成的簡單命令

匯入NumPy

import numpy as np

建立全為0的陣列

np.zeros((3,3))

建立全為1的陣列

np.ones((2,3,4))

建立單位矩陣

np.eyes(3)

一維等差陣列

np.arange(6)

轉換成二維

np.arange(6).reshape(2,3)

一維等間隔陣列

np.linspace(1, 10, num=6)

隨機陣列

np.random.rand(2,3)

隨機整數陣列(十以內)

np.random.randint(10, size=(2,3))

自定義函式建立陣列

np.fromfunction(lambda i, j: i + j, (3, 3))

輸出結果為array([[0,1,2],[1,2,3],[2,3,4]])