python - 将 1d 数组添加到 3d 数组的每个 2d 数组上

Question

假设我有以下大小为 (2,3,2) 的数组 a 和大小为 (2) 的数组 b。

import numpy as np

a = np.array([[[1, 2], [3, 4], [5, 6]], [[7, 8], [9, 10], [11, 12]]])
b = np.array([0.2, 0.8])

数组a看起来像这样:

我想使用 numpy 例程将 b 连接到 a 中每个二维数组的第一行以生成数组

我似乎无法使concatenate、vstack、append等工作。

Answer 1

试试这个:

np.concatenate(([[b]]*2,a),axis=1)
# Result:
array([[[  0.2,   0.8],
        [  1. ,   2. ],
        [  3. ,   4. ],
        [  5. ,   6. ]],

       [[  0.2,   0.8],
        [  7. ,   8. ],
        [  9. ,  10. ],
        [ 11. ,  12. ]]])