numpy实现基础的矩阵加减乘除

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