Numpy使用

1 导入模块

import numpy as np

2 多维数组

a = np.array([1,2,3,4]) #生成一维数组
np.ndim(a) #查看数组的维数  # 1
a.shape #查看数组的形状，返回结果是元组  # (4,)
a.shape[0] # 4

B = np.array([[1,2],[3,4],[5,6]]) #生成二维数组
np.ndim(B) # 2
B.shape #（3,2)

3 矩阵乘法

np.dot(A,B) #矩阵相乘

4 获取最值索引

注：矩阵的第0维是列方向，第1维是行方向

np.argmax(x) #获取X中所有元素最大值索引
np.argmin(x) #获取X中所有元素最小值索引
np.argmax(x,axis=1) #如果x是二维数组，将找出每行的最大值素引