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

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

Python
1numpy.array([
2    [
3        [0],
4        [0],
5    ],
6    [
7        [1],
8        [1],
9    ]
10])
11
12# 以下のような形にキャスト
13numpy.array([
14    [
15        [0, 0, 0],
16        [0, 0, 0],
17    ],
18    [
19        [1, 1, 1],
20        [1, 1, 1],
21    ]
22])
23

行動規範の内容に同意します

回答2件

numpy.broadcast_to — NumPy Manual

python
1import numpy as np
2
3arr = np.array([[[0], [0]], [[1], [1]]])
4
5new_shape = (*arr.shape[:-1], 3)
6np.broadcast_to(arr, new_shape)
7# array([[[0, 0, 0],
8#         [0, 0, 0]],
9# 
10#        [[1, 1, 1],
11#         [1, 1, 1]]])

投稿2021/04/05 00:50

kirara0048

総合スコア1399

teefpc

2021/04/05 10:59

ありがとうございます。

行動規範の内容に同意します

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

python
1>>> print(a)
2[[[0]
3  [0]]
4
5 [[1]
6  [1]]]
7>>>
8>>> b = np.concatenate([a, a, a], axis=2)
9>>>
10>>> print(b)
11[[[0 0 0]
12  [0 0 0]]
13
14 [[1 1 1]
15  [1 1 1]]]