zl程序教程

您现在的位置是:首页 >  其他

当前栏目

numpy矩阵操作

2023-04-18 12:44:00 时间

广播

for循环计算

a = np.array([1,2,3,4]) 
b = np.array([10,20,30,40]) 
c = a * b 

输出[ 10 40 90 160]

切片

取值[0,0],[1,1],[2,0]

import numpy as np 
 
x = np.array([[1,  2],  [3,  4],  [5,  6]]) 
y = x[[0,1,2],  [0,1,0]]  
print (y)
  • start: end:step 切片范围,end默认-1
  • [1,2] 切片索引
  • … 取所有

向量计算

  • dot对应的索引相乘
  • vdot 向量点积
  • matmul矩阵相乘