通过将 function 应用于 nx 1 numpy 数组中的元素对，numpy 中的 nxn 矩阵

Question

如果我在 numpy a中有一个数组，它是nx 1 。 此外，我有一个 function F(x,y) ，它接受两个值并返回一个值。 我想构造一个nxn矩阵b ，其中 b_ij = F(a_i, a_j) （在数组a中）。 有没有办法在不循环遍历 arrays 的情况下执行此操作？

Answer 1

假设你的 function 是：

def F(a_i, a_j):
    return (a_i + a_j) if a_i % 2 == 0 else (a_i + a_j + 1)

要在 2 arrays in 1 go 上调用它，请定义此 function 的矢量化版本：

FF = np.vectorize(F)

然后调用它：

result = FF(a, a.T)

作为我使用的源数组：

a = np.array([[1], [5], [10], [50], [80]])

所以它的形状是(5, 1) （单列数组）并得到：

array([[  3,   7,  12,  52,  82],
       [  7,  11,  16,  56,  86],
       [ 11,  15,  20,  60,  90],
       [ 51,  55,  60, 100, 130],
       [ 81,  85,  90, 130, 160]])