numpy中的运算符重载

下面统计了numpy中的运算符重载及其功能。

+

=np.add()
功能：对位相加

a = np.array([[1, 2]])
b = np.array([[3, 4]])
print(a+b)
# [[4 6]]

* 同样可用于yinshi类型转换

-

=np.subtract()
功能：对位相减

print(b-a)
# [[2 2]]

*

=np.multiply()
功能：对位相乘

print(a*b)
# [[3 8]]

/

=np.divide()
功能：对位相除

print(a/b)
# [[0.33333333 0.5       ]]

@

=np.matmul()
功能：矩阵相乘

print(a.T @ b)
# [[3 4]
#  [6 8]]

> < ==

功能：对位比较，返回相同shape的矩阵

print(a>b)
# [[False False]]

# 经常与broadcasting一起使用：
D = np.array([[0.45, 0.55],[0.85, 0.65]])
print((D > 0.5) + 0)
# [[0 1]
#  [1 1]]