0Pricing
Learn AI with Python · 课时

广播与向量化运算

了解广播的工作原理、逐元素运算,以及如何通过向量化避免循环

广播与向量化运算 是 CoddyKit 上的免费 Learn AI with Python 课时。 这是第 3 节课,共 4 节。 你可以在下方免费阅读本课时的完整内容 — 然后在浏览器中使用内置代码编辑器和全天候 AI 导师进行实践。 这是 Learn AI with Python 学习路径的一部分,你的进度在网页和 CoddyKit 应用中同步。 Learn AI with Python 课程共包含 4 节课。

为什么要向量化

NumPy 会以 C 的速度处理整个数组。将 Python 循环替换为数组表达式,这种做法称为向量化,通常能提速 10 到 100 倍。

逐元素运算

当形状匹配时,算术运算符会逐个元素地应用。无需编写循环。

import numpy as np
a = np.array([1, 2, 3])
b = np.array([10, 20, 30])
print(a + b)   # [11 22 33]
print(a * b)   # [10 40 90]

标量广播

数组与标量之间的运算会将该标量应用到每个元素。这个标量会在整个数组中进行“广播”。

a = np.array([1, 2, 3, 4])
print(a * 10)   # [10 20 30 40]
print(a + 100)  # [101 102 103 104]

什么是广播

广播允许 NumPy 组合形状不同的数组:它会在逻辑上拉伸较小的数组,而不会复制内存。

广播规则

请从末尾(最右侧)维度开始比较形状。当两个维度相等,或其中一个维度为 1 时,它们是兼容的。大小为 1 的维度会拉伸以匹配另一个维度。

# (3, 4) with (4,)   -> compatible (trailing 4 matches)
# (3, 4) with (3, 1)  -> compatible (1 stretches to 4)
# (3, 4) with (3,)    -> ERROR (trailing 4 vs 3)

行向量广播

长度等于列数的一维数组会加到每一行上。

m = np.array([[1, 2, 3], [4, 5, 6]])
row = np.array([10, 20, 30])
print(m + row)
# [[11 22 33]
#  [14 25 36]]

列向量广播

请将向量重塑为列向量(形状为 (n,1)),这样它就会沿列方向而不是行方向进行广播。

col = np.array([[100], [200]])   # shape (2, 1)
print(m + col)
# [[101 102 103]
#  [204 205 206]]

数据居中(实际应用)

减去每列的 mean 可以使每个特征居中,这是常见的预处理步骤,只需一行广播代码即可完成。

data = np.array([[1.0, 10.0], [2.0, 20.0], [3.0, 30.0]])
centered = data - data.mean(axis=0)
print(centered)
# columns now have mean 0

通用函数

NumPy 提供了称为通用函数的快速逐元素数学函数:np.sqrt、np.log、np.exp 等。它们支持向量化,也支持广播。

a = np.array([1.0, 4.0, 9.0, 16.0])
print(np.sqrt(a))   # [1. 2. 3. 4.]
print(np.log(a))    # natural log of each element
print(np.exp([0, 1]))  # [1.        2.71828183]

循环与向量化速度

同一计算经过向量化后会快得多。请使用 %timeit 在一百万个元素上感受差异。

big = np.arange(1000000, dtype=float)
# Slow Python loop:
# %timeit [x*x for x in big]
# Fast vectorized:
%timeit big * big   # typically tens of times faster

组合向量化

将通用函数和算术运算链接起来,可以直接表达公式。下面的计算无需循环即可得到高斯曲线。

x = np.linspace(-3, 3, 7)
y = np.exp(-x**2 / 2)
print(np.round(y, 3))

快速检查

请测试您对广播的理解。

回顾

向量化要点:

  • 逐元素运算无需 Python 循环,并以 C 的速度运行
  • 广播会对齐末尾维度;各维度大小必须匹配或为 1
  • 使用列向量 (n,1) 让数据沿列方向广播
  • 通用函数:np.sqrt、np.log、np.exp
  • 使用 data - data.mean(axis=0) 将数据居中

常见问题解答

「广播与向量化运算」课时是免费的吗?

是的 — 「广播与向量化运算」的完整文本可在网页上免费阅读。要进行交互式练习(内置代码编辑器和全天候 AI 导师)并解锁 Learn AI with Python 课程的其余内容,请升级到 CoddyKit PRO。 Learn AI with Python 课程共包含 4 节课。

「广播与向量化运算」这节课中我会学到什么?

了解广播的工作原理、逐元素运算,以及如何通过向量化避免循环 你通过在浏览器中直接运行的动手代码来练习 Learn AI with Python,全天候 AI 导师会在你学习这节课的过程中回答你的问题。

学习 Learn AI with Python 需要有经验吗?

无需任何先前经验。CoddyKit 上的 Learn AI with Python 课程适合初学者到高级学习者,你可以从这里开始或从头开始,按照自己的节奏学习。 这是第 3 节课,共 4 节。

「广播与向量化运算」课时需要多长时间?

大多数 CoddyKit 课程大约需要 5–10 分钟。每节课都很精短且互动,所以你能稳步进步,并在网页和应用中从离开的地方继续。

我能在这节 Learn AI with Python 课中编写并运行代码吗?

能。每节 Learn AI with Python 课都包含内置代码编辑器,你可以在浏览器中直接编写并运行真实代码,并获得即时 AI 反馈 — 无需本地设置。

此课程中的所有课时

  1. 数组创建与属性
  2. 索引、切片与花式索引
  3. 广播与向量化运算
  4. 使用 NumPy 进行线性代数运算
← 返回 Learn AI with Python