Python: 将对应于 1d 数组的 2d numpy 数组相乘

当我们需要将 1 维数组（一个向量）与 2 维数组（一个矩阵）相乘时，我们需要用到 numpy 库中的 dot() 函数。dot() 函数可以计算两个数组的点积。

请看下面的代码示例：

import numpy as np

a = np.array([1, 2, 3]) # 一维数组
b = np.array([[1, 2], [3, 4], [5, 6]]) # 二维数组

c = np.dot(a, b)

print(c)

运行结果为：

[22 28]

此处，a 是一个一维数组，长度为 3；b 是一个二维数组，大小为 3 行 2 列。

使用 dot() 函数，我们可以对 a 和 b 进行矩阵乘法，得到一个大小为 1 行 2 列的结果数组。