NumPy数组乘法详解：区分元素级运算与点积

本教程深入探讨numpy数组的乘法机制，重点区分了元素级乘法（`*`运算符）与矩阵点积（`np.dot()`和`np.matmul()`）。文章通过具体示例，解析了不同乘法操作的行为差异，强调了数组维度对结果的影响，并指导读者如何通过`reshape`操作实现期望的矩阵乘法结果，确保在数据科学和机器学习中正确应用numpy乘法。

1. 引言：NumPy数组乘法的核心概念

NumPy是Python中用于科学计算的核心库，提供了高性能的多维数组对象和各种派生工具。在NumPy中，数组乘法是一个基础且频繁使用的操作，但其行为往往因操作符的选择和数组维度的差异而产生混淆。本文旨在深入解析NumPy中两种主要的乘法类型：元素级乘法和矩阵点积，并通过实例演示它们的区别与正确用法。理解这些概念对于高效且准确地进行数据处理和数值计算至关重要。

2. 元素级乘法 (* 运算符)

当使用标准的乘法运算符*对NumPy数组进行操作时，执行的是元素级（element-wise）乘法。这意味着对应位置上的元素会进行相乘，要求参与运算的数组形状兼容（通过广播机制）。

示例分析：

考虑以下两个NumPy数组：

• a = np.array([1, 2, 3])，其形状为 (3,)，表示一个一维数组。
• b = np.array([[1]])，其形状为 (1,1)，表示一个二维数组（1行1列）。

当我们执行 a * b 时，NumPy会应用其广播（Broadcasting）规则来使数组形状兼容。在这个例子中：

1. 数组 a ((3,)) 首先被视为 (1,3)。
2. 数组 b ((1,1)) 会沿着其第二个维度（列）广播，复制其内容以匹配 a 的形状。
3. 最终，广播后的 a 变为 [[1, 2, 3]]，广播后的 b 变为 [[1, 1, 1]]。
4. 然后执行元素级乘法 [[1, 2, 3]] * [[1, 1, 1]]，结果自然是 [[1, 2, 3]]。

代码示例：

import numpy as np

a = np.array([1, 2, 3])
b = np.array([[1]])

print(f"数组 a: {a}, 形状: {a.shape}")
print(f"数组 b: {b}, 形状: {b.shape}")

# 执行元素级乘法
result_element_wise = a * b
print(f"\n使用 * 运算符进行元素级乘法的结果:\n{result_element_wise}")
print(f"结果形状: {result_element_wise.shape}")

输出：

数组 a: [1 2 3], 形状: (3,)
数组 b: [[1]], 形状: (1, 1)

使用 * 运算符进行元素级乘法的结果:
[[1 2 3]]
结果形状: (1, 3)

3. 矩阵点积 (np.dot() 和 np.matmul())

如果你的目标是执行线性代数中的矩阵乘法（也称为点积），则应该使用 np.dot() 或 np.matmul() 函数。这些函数遵循矩阵乘法的数学规则，对输入数组的维度有严格的要求。

新CG儿

数字视觉分享平台 | AE模板_视频素材

412

查看详情

矩阵乘法规则回顾： 如果矩阵 A 的形状是 (m, n)，矩阵 B 的形状是 (n, p)，那么它们的乘积 C = A @ B 的形状将是 (m, p)。关键在于 A 的列数必须等于 B 的行数。

实现期望的点积结果：

用户在原始问题中期望得到 [[1],[2],[3]]。这实际上是希望将 a 视为一个 (3,1) 的列向量，与 b (一个 (1,1) 的矩阵) 进行矩阵乘法。

要实现这个结果，我们需要首先将数组 a 的形状从 (3,) 转换为 (3,1)。这可以通过 reshape() 方法或 np.newaxis 来完成。

1. 重塑 a： 将 a 重塑为 (3,1)。 a_reshaped = a.reshape(3, 1) 或者 a_reshaped = a[:, np.newaxis]

2. 执行点积： 现在 a_reshaped 的形状是 (3,1)，b 的形状是 (1,1)。它们满足矩阵乘法规则（a_reshaped 的列数 1 等于 b 的行数 1）。 result_dot_product = np.dot(a_reshaped, b)

代码示例：

import numpy as np

a = np.array([1, 2, 3])
b = np.array([[1]])

# 将 a 重塑为 (3, 1) 的列向量
a_reshaped = a.reshape(3, 1)
# 或者使用 np.newaxis: a_reshaped = a[:, np.newaxis]

print(f"\n重塑后的数组 a_reshaped: \n{a_reshaped}, 形状: {a_reshaped.shape}")
print(f"数组 b: \n{b}, 形状: {b.shape}")

# 使用 np.dot() 进行矩阵点积
result_dot_product = np.dot(a_reshaped, b)
print(f"\n使用 np.dot() 进行矩阵点积的结果:\n{result_dot_product}")
print(f"结果形状: {result_dot_product.shape}")

# np.matmul() 也可以实现相同的效果
result_matmul = np.matmul(a_reshaped, b)
print(f"\n使用 np.matmul() 进行矩阵点积的结果:\n{result_matmul}")
print(f"结果形状: {result_matmul.shape}")

输出：

重塑后的数组 a_reshaped:
[[1]
 [2]
 [3]], 形状: (3, 1)
数组 b:
[[1]], 形状: (1, 1)

使用 np.dot() 进行矩阵点积的结果:
[[1]
 [2]
 [3]]
结果形状: (3, 1)

使用 np.matmul() 进行矩阵点积的结果:
[[1]
 [2]
 [3]]
结果形状: (3, 1)

关于 np.dot() 和 np.matmul() 的细微差别：

• 对于二维数组，np.dot() 和 np.matmul() 的行为是相同的。
• 对于一维数组，np.dot() 执行的是向量内积（返回一个标量），而 np.matmul() 会尝试将一维数组提升为矩阵（例如，[1,2,3] 视为 [[1,2,3]] 或 [[1],[2],[3]]，取决于上下文），然后执行矩阵乘法。
• 对于更高维的数组，np.matmul() 遵循“堆叠矩阵”乘法规则，而 np.dot() 则更为通用，可以处理更复杂的轴求和。在大多数常见的矩阵乘法场景中，np.matmul()（或 @ 运算符）是更直观的选择。

4. 注意事项与最佳实践

1. 明确乘法意图： 在进行NumPy数组乘法时，首先要明确你想要执行的是元素级乘法还是矩阵点积。这决定了你选择的操作符或函数。
2. 点积优先使用专用函数： 即使在某些特定情况下，通过巧妙的广播，* 运算符也能在重塑后得到与点积相同的结果（如本例中 a_reshaped * b），但为了代码的清晰性、可读性和通用性，始终推荐使用 np.dot() 或 np.matmul() 来执行矩阵点积。

   # 示例：重塑后 a_reshaped * b 的结果
   a = np.array([1, 2, 3])
   b = np.array([[1]])
   a_reshaped = a.reshape(3, 1)
   result_element_wise_after_reshape = a_reshaped * b
   print(f"\n重塑后 a_reshaped * b 的结果:\n{result_element_wise_after_reshape}")
   print(f"结果形状: {result_element_wise_after_reshape.shape}")
   print("注意：虽然在这个特定例子中结果相同，但推荐使用np.dot()或np.matmul()进行点积运算。")

   登录后复制
3. 检查数组形状： 在执行任何乘法操作之前，养成检查数组形状的好习惯（使用 array.shape 属性）。这有助于预判操作结果，并在出现维度不匹配时快速定位问题。
4. 灵活运用 reshape() 和 np.newaxis： 当数组的当前形状不满足你的乘法需求时，reshape() 方法和 np.newaxis（用于增加维度）是调整数组维度的强大工具。

5. 总结

NumPy数组的乘法操作看似简单，实则蕴含着元素级运算与矩阵点积的深刻区别。* 运算符执行的是元素级乘法，依赖于NumPy的广播机制；而 np.dot() 和 np.matmul() 则用于执行严格的矩阵点积，遵循线性代数的规则。理解这两种乘法的本质差异，并根据具体的计算需求选择正确的函数和操作符，是掌握NumPy高级用法的基石。同时，熟练运用数组重塑技巧和始终关注数组维度，将帮助你避免常见的错误，并编写出高效、健壮的NumPy代码。

以上就是NumPy数组乘法详解：区分元素级运算与点积的详细内容，更多请关注php中文网其它相关文章！