4

私は次元を持つnumpy配列を持っています12 x 12 x 4。今、この立方体にレイヤーを追加して、12 x 13 x 4配列を作成しようとしています。この 13 番目のレイヤーには、最初の軸からの対応するインデックスが含まれている必要があるため、たとえばアドレス指定の[7, 13, :]結果は[7, 7, 7, 7].

説明するのは難しいですが、誰かがnumpyでこれを達成する方法についてアドバイスを持っていますか?

編集:少し複雑すぎるようですが、解決策を見つけました:

# Generate extra layer
layer = np.repeat(np.arange(0, 12)[:, np.newaxis], data.shape[2], axis=1)

# Get dimensions right...
layer = np.expand_dims(layer, axis=1)

# ... and finally append to data
result = np.append(data, layer, axis=1)

より良い提案のためにまだ開いています。

4

1 に答える 1