python - 如何向 NumPy 数组添加多个额外的列

Question

假设我有两个 NumPy 数组，a 和 b:

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

b = np.array([8,9])

我想将相同的数组 b 附加到每一行(即添加多列)以获得一个数组，c:

b = np.array([
    [1, 2, 3, 8, 9],
    [2, 3, 4, 8, 9]
    ])

如何在 NumPy 中轻松高效地执行此操作？

我特别担心它在大数据集上的行为(其中 a 比 b 大得多)，是否有任何方法可以创建许多副本(即 a.shape[0]) b?

与this question相关，但具有多个值。

Answer 1

这是一种方法。我认为它是有效的，因为它是矢量化的。它依赖于这样一个事实，即在矩阵乘法中，用列 (1, 1) 预乘一行将产生该行的两个堆叠副本。

import numpy as np

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

b = np.array([[8,9]])

np.concatenate([a, np.array([[1],[1]]).dot(b)], axis=1)

Out: array([[1, 2, 3, 8, 9],
            [2, 3, 4, 8, 9]])

请注意，b 的指定略有不同(作为二维数组)。

Is there any way around creating many copies of b?

最终结果包含这些副本(而 numpy 数组实际上是内存中的值数组)，所以我不明白如何。