numpy数组的广播机制

数组的广播

• 当一个数组进行运算的时候，整个数组都会进行运算，称为广播

1：数组与一个数字的广播

• 一个数组可以和一个数字进行加、减、乘、除、取余、平方等运算
• 元素不会改变

代码演示

import numpy as np

# 数组
a = np.random.randint(100, 200, size=(3, 3))

# 数字
b = 5

# 加运算
a + b

# 减运算
a - b

# 乘运算
a * b

# 除运算
a / b

# 整除运算
a // b

# 平方运算
a ** 2

# 取余运算
a % b

运行效果

2：高维数组对低维数组的广播

• 数组对数组广播的前提
  • 1：低纬度数组形状，必须要等于高维度数组的一部分形状，要不然无法进行广播
  • 2：数组的广播只能对，X轴，Y轴，Z轴方向进行广播，不能斜着广播
  • 3：数组的广播同样可以进行四则运算

代码演示

# 一维数组对二维数组的广播
a  = random.randint(10, 20, size=(1, 3))
b = random.randint(10, 20, size=(3, 3))
# 加运算
b + a
# 减运算
b - a
# 乘运算
b * a
# 除运算
b / a
# 整除运算
b // a
# 平方运算
b ** 2
# 取余运算
b % a

# 对数组进行转置操作
c = b.T

运行效果

• 转置后在进行广播
  

数组的拼接

• 数组的拼接式两个形状形似的数组的拼接

数组的左右拼接

• 两个数组的左右拼接

代码演示

import numpy as np

a = np.random.randint(10, 20, size=(3, 3))
b = np.random.randint(20, 30, size=(3, 1))
c = np.hstack([a, b])  # 注意传入参数的先后顺序会对结果有影响

数组的上下拼接

代码演示

a17 = np.random.randint(10, size=(3, 3))
a18 = np.random.randint(10, size=(3,))
a19 = np.vstack([a17, a18])  # 注意传入参数的先后顺序会对结果有影响

数组的拼接

代码演示

a17 = np.random.randint(10, size=(3, 3))
a18 = np.random.randint(10, size=(3,))
np.concatenate((a27, a25), axis=0)  # 按行进行拼接
np.concatenate((a27, a25), axis=1)  # 按列进行拼接，注意这个地方可能需要使用一个数组的转置