Python中对numpy库的学习（四）二元函数运算、广播以及自定义ufunc函数

一、二元函数

概念：二元函数是参数为两个数组的函数，包括算数运算和布尔运算。

1、算术运算

以下代码为两个形状相同的函数进行运算(布尔运算也是)，运算规则是对应位置进行四则运算。

import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
print(a+b)      # [5 7 9]
print(a-b)      # [-3 -3 -3]
print(a*b)      # [ 4 10 18]
print(b/a)      # [4.  2.5 2. ]
print(b//a)     # [4 2 2] 整除
print(b % a)    # [0 1 0] 取余

2、布尔运算

布尔运算是指根据使用">"，"<"，"<="，">="，“==”，"!="等逻辑运算比较两个数组，并返回布尔型数组

a = np.array([1, 2, 3])
b = np.array([1, 5, 2])
print(a < b)    # [False  True False]
print(a == b)   # [ True False False]

二、广播

当数组形状不同时，低维的数组与高维数组进行运算时，会自动扩展对齐高维数组进行运算。这种从低维向高维自动扩展被称为广播。

1、最常见的广播方式

一个数组+一个数字