矢量矢量乘法创建矩阵

Question

我是一个慢慢切换到numpy / scipy的IDL用户，并且我在IDL中经常执行一项操作，但无法通过numpy重现：

IDL> a = [2., 4]
IDL> b = [3., 5]
IDL> print,a # b
      6.00000      12.0000
      10.0000      20.0000

我甚至不确定这个操作的名称（英语不是我的主要语言）。也许很明显如何在numpy中做到这一点，但我找不到一个简单的方法。

感谢。

- 亚瑟;

Answer 1

这被称为两个向量的outer product。您可以使用np.outer：

import numpy as np

a = np.array([2, 4])
b = np.array([3, 5])
c = np.outer(a, b)

print(c)
# [[ 6 10]
#  [12 20]]

假设您的两个输入都是numpy数组（而不是Python列表等），您也可以使用broadcasting标准*运算符：

# you could also replace np.newaxis with None for brevity (see below)
d = a[:, np.newaxis] * b[np.newaxis, :]

您还可以将np.dot与广播结合使用：

e = np.dot(a[:, None], b[None, :])

另一个鲜为人知的选择是使用np.multiply ufunc的.outer方法：

f = np.multiply.outer(a, b)

我个人会使用np.outer或*进行广播。