numpy和pytorch基本語法差不多

匯入

import numpy as np

建立變數

手動生成

a=np.array([[1,2,3],[4,5,6]])
a=np.array[]

自動生成

np.ones((3,4))
np.zeros((2,3))：0
np.empty((3,2))：無限接近0
np.random.random((3,2)) 生成3行2列從0到1的隨機數
np.random.normal(size=(3,2))#生成3行2列符合標準正態分佈的隨機數
np.random.randint(0,10,size=(3,2))#生成3行2列從0到10的隨機整數

查詢

型別：a.dtype

形狀

變形：a.reshape()
陣列形狀：a.shape 沒有a.size()
維度：a.ndim
大小：a.size #長度
變一維，推平：a.flatten()

基本運算

-
* a.dot()行乘列
//
%
**

>:
arr3 = arr1 > 3：
[[False False False]
[ True True True]]

函式

np.sum(sample1)#求和
np.min(sample1)#求最小值
np.max(sample1)#求最大值
np.sum(sample1,axis=0)#對列求和
np.sum(sample1,axis=1)#對行求和
np.sum(sample1,axis=0)#對列求和
sample1.argmin()#索引
sample1.mean()#求平均值
np.median(sample1)#求中位數
np.sqrt(sample1)#開方

轉置

a.T
a.transpose()

排序

np.sort(sample4)#排序
np.clip(sample4,2,7)#小於2就變成2，大於7就變為7

索引

和pytorch差不多