使用 numpy 的常见操作

  # 导入numpy库
  import numpy as np
  
  # 1. 数组
  # 1.1 创建一个一维数组
  arr = np.array([1,2,3])
  
  # 1.2 创建一个多维数组
  arr = np.array([[1,2], [3,4]])
  
  # 1.3 查看数组形状和类型
  print(arr.shape)
  print(arr.dtype)
  
  # 2. 数值运算
  # 2.1 加法
  result = arr + np.array([5,6])
  
  # 2.2 减法
  result = arr - np.array([5,6])
  
  # 2.3 乘法
  result = arr * np.array([5,6])
  
  # 2.4 除法
  result = arr / np.array([5,6])
  
  # 2.5 指数运算
  result = np.exp(arr)
  
  # 3. 索引和切片
  # 3.1 通过索引访问元素
  print(arr[0])
  
  # 3.2 通过切片获取子数组
  result = arr[:5]
  
  # 4. 线性代数运算
  # 4.1 矩阵运算
  A = np.array([[1,2], [3,4]])
  B = np.array([[5,6], [7,8]])
  C = np.dot(A,B)
  
  # 4.2 线性代数运算
  result = np.linalg.inv(A)
  
  # 5. 其他功能
  # 5.1 随机生成数组
  arr = np.random.randint(0,10,(3,4))
  
  # 5.2 转置数组
  result = arr.T
  
  # 5.3 保存数组
  np.save('arr.npy', arr)