Adicionando colunas extras a um array NumPy

Suponha que você tenha um array NumPy 2D da seguinte maneira:

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

Para adicionar uma coluna de zeros ao longo do segundo eixo, você pode utilizar vários métodos. Uma abordagem é empregar a função np.c_[ ]:

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

Isso criará um novo array b com uma coluna adicional de zeros:

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

Como alternativa, você pode usar a função np.r_[ ]:

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

Este método também adicionará uma coluna de zeros ao array.

Observe que np.r_[ ] e np.c_[ ] fornecem opções flexíveis para modificar as dimensões do array. Eles podem ser utilizados para misturar vetores e escalares, adicionar linhas ou colunas e até inserir arrays inteiros em posições especificadas.