numpyで特定の次元に要素を追加したい

Question

numpyで特定の次元に要素を追加したいです。
具体的には(2,2,1)の形状を(2,2,3)にキャストしたいです。

numpy.array([
    [
        [0],
        [0],
    ],
    [
        [1],
        [1],
    ]
])

# 以下のような形にキャスト
numpy.array([
    [
        [0, 0, 0],
        [0, 0, 0],
    ],
    [
        [1, 1, 1],
        [1, 1, 1],
    ]
])

Answer 1

numpy.broadcast_to — NumPy Manual

import numpy as np

arr = np.array([[[0], [0]], [[1], [1]]])

new_shape = (*arr.shape[:-1], 3)
np.broadcast_to(arr, new_shape)
# array([[[0, 0, 0],
#         [0, 0, 0]],
# 
#        [[1, 1, 1],
#         [1, 1, 1]]])

Comments

[teefpc]

ありがとうございます。

Answer 2

要素数が違いますので、形状の変更ではなく、新しいndarrayを作ることになります。

>>> print(a)
[[[0]
  [0]]

 [[1]
  [1]]]
>>>
>>> b = np.concatenate([a, a, a], axis=2)
>>>
>>> print(b)
[[[0 0 0]
  [0 0 0]]

 [[1 1 1]
  [1 1 1]]]

Comments

[teefpc]

ありがとうございます。