numpy简单学习

import numpy as np

if __name__ == '__main__':
    #np.dtype 矩阵数据类型，np.shape (3，3）3乘3矩阵，np.ndim 矩阵维数
    a = np.array([[1,2,3],[1,2,3],[4,5,6]])

    np.uint8([1])
    np.arange(2,10,2)  #[2,10)间距为2的矩阵
    np.linspace(0,2*np.pi,5)   #[0.         1.57079633 3.14159265 4.71238898 6.28318531]一共5个数
    b=np.zeros((4,3),dtype=np.int32) #零矩阵，4乘3，int型
    np.ones((2,2),dtype=np.float64)   #全1矩阵
    np.eye(2)   #2乘2单位矩阵
    np.identity(3)   #3乘3单位矩阵
    b = np.random.randint(1,6,(3,3),dtype=np.uint8)  #1到6范围内float型2乘3矩阵
    print(b)

    #矩阵变换
    a = a.reshape((1,9))    #[[1 2 3 1 2 3 4 5 6]]
    a = a.reshape((-1,3))   #必须3列
    a = a.flatten()   #转为一维数组[1 2 3 1 2 3 4 5 6]

    b = b.T  #转置
    c = b.transpose()  #转置
    #print(b)
    a = np.array([[1, 2, 3], [1, 2, 3], [4, 5, 6]])
    b = np.array([[1,1,1],[2,2,2],[3,3,3]])
    d = np.hstack((a,b,a))   #水平拼接,维数必须相同
    e = np.vstack((a,b,a))   #竖直拼接

    print(a.max()) #最大值np.max(a)
    a.min()   #最小值
    a.mean()  #均值
    A = a.max(axis=0)  #每一列的最大值[4 5 6]
    B = a.max(axis=1)  #每一行的最大值[3 3 6]
    # a = a + b
    # print(a)  #[[2 3 4][3 4 5][7 8 9]]
    # np.power(a,2)   #每项都2次幂
    # np.sqrt(b)   #每项都开方
    # np.log(a)
    # np.log10(b)
    # np.log2(a)
    C = a * b  #每项相乘[[ 1  2  3][ 2  4  6][12 15 18]]
    D = a @ b   #a的行乘b的列，矩阵乘法[[14 14 14][14 14 14][32 32 32]]
    a.dot(b)   #同 D = a @ b

    print("a的第0行：%s" %a[0,:])
    print("a的第0列：%s"  %a[:, 0])
    print("a的前两行：%s" %a[0:2, :])   #[0,2)不包含第三行
    print("a的右下角：%s" %a[1:, 1:])
    a[0,0] = 10
    a[0,:] = np.array([4,5,6])
    # a的第0行：[1 2 3]
    # a的第0列：[1 1 4]
    # a的前两行：[[1 2 3]
    #        [1 2 3]]
    # a的右下角：[[2 3]
    #        [5 6]]