python - Numpy で 1 次元配列を 3 次元配列に追加する

Question

2 つの配列を追加しようとしています。

np.zeros((6,9,20)) + np.array([1,2,3,4,5,6,7,8,9])

みたいなものを出したい

array([[[ 1.,  1.,  1., ...,  1.,  1.,  1.],
        [ 2.,  2.,  2., ...,  2.,  2.,  2.],
        [ 3.,  3.,  3., ...,  3.,  3.,  3.],
        ..., 
        [ 7.,  7.,  7., ...,  7.,  7.,  7.],
        [ 8.,  8.,  8., ...,  8.,  8.,  8.],
        [ 9.,  9.,  9., ...,  9.,  9.,  9.]],

       [[ 1.,  1.,  1., ...,  1.,  1.,  1.],
        [ 2.,  2.,  2., ...,  2.,  2.,  2.],
        [ 3.,  3.,  3., ...,  3.,  3.,  3.],
        ..., 
        [ 7.,  7.,  7., ...,  7.,  7.,  7.],
        [ 8.,  8.,  8., ...,  8.,  8.,  8.],
        [ 9.,  9.,  9., ...,  9.,  9.,  9.]],

したがって、対応する列の各マトリックスにエントリを追加します。ある種のループでコーディングできることはわかっていますが、よりエレガントで高速なソリューションを使用しようとしています。

Answer 1

またはbroadcastingを使用して 2 番目の配列の次元を拡張した後、次のように機能させることができます-Nonenp.newaxis

np.zeros((6,9,20))+np.array([1,2,3,4,5,6,7,8,9])[None,:,None]