numpy实现基础的矩阵加减乘除
import numpy as np
#定义两个矩阵
A = np.array([[1,2],[3,4]], dtype = np.float64)
B = np.array([[5,6],[7,8]], dtype = np.float64)
#矩阵元素相加
# print(A+B)
print("A+B:",np.add(A,B))
#矩阵元素相减
# print(B-A)
print("B-A:",np.subtract(B,A))
#矩阵元素相乘
# print(A*B)
print("A*B:",np.multiply(A,B))
#矩阵点乘
# print(A.dot(B))
print("A.*B:",np.dot(A,B))
#矩阵元素相除
# print(A/B)
print("A/B:",np.divide(A,B))
#矩阵每个元素开方
print("A开方:",np.sqrt(A))
#矩阵的转置
print("A的转置:",A.T)
A+B: [[ 6. 8.] [10. 12.]] B-A: [[4. 4.] [4. 4.]] A*B: [[ 5. 12.] [21. 32.]] A.*B: [[19. 22.] [43. 50.]] A/B: [[0.2 0.33333333] [0.42857143 0.5 ]] A开方: [[1. 1.41421356] [1.73205081 2. ]] A的转置: [[1. 3.] [2. 4.]]
这里的dot是计算向量内积的
C = np.array([9,10], dtype = np.float64)
D = np.array([1,2], dtype = np.float64)
print("A*C:",np.dot(A,C))
print("C*D:",np.dot(C,D))
A*C: [29. 67.] C*D: 29.0