NumPy 配列に追加の列を追加する

次のような 2D NumPy 配列 a があるとします。

a = np.array([
    [1, 2, 3],
    [2, 3, 4],
])

2 番目の軸に沿ってゼロの列を追加するには、さまざまな方法を利用できます。 1 つのアプローチは、np.c_[ ] 関数を使用することです:

b = np.c_[a, np.zeros(a.shape[0])]

これにより、新しい追加のゼロ列を持つ配列 b:

b = np.array([
    [1, 2, 3, 0],
    [2, 3, 4, 0],
])

または、np.r_[ ] 関数を使用することもできます:

b = np.r_[a, np.zeros((a.shape[0], 1))]

このメソッドは、配列にゼロの列も追加します。

np.r_[ ] と np.c_[ ] には、配列の次元を変更するための柔軟なオプションが用意されていることに注意してください。これらを利用して、ベクトルとスカラーを混合したり、行や列を追加したり、配列全体を指定した位置に挿入したりすることもできます。