Adicionar linha à matriz em NumPy

Manav Narula 30 janeiro 2023
  1. Use a função numpy.vstack() para adicionar uma linha a um array em NumPy
  2. Use a função numpy.append() para adicionar uma linha a um array em NumPy
  3. Use a função numpy.r_() para adicionar uma linha a um array em NumPy
  4. Use a função numpy.insert() para adicionar uma linha a um array em NumPy
Adicionar linha à matriz em NumPy

As matrizes são frequentemente usadas em matemática e estatística para representação de dados e resolução de várias equações lineares. Na programação, um array bidimensional é tratado como um array.

Em Python, o módulo numpy é usado para trabalhar com matrizes. Ele tem muitas funções e classes disponíveis para realizar diferentes operações em matrizes.

Neste tutorial, aprenderemos como adicionar uma linha a um array em numpy.

Use a função numpy.vstack() para adicionar uma linha a um array em NumPy

A função vstack() empilha as matrizes verticalmente. Empilhar duas matrizes 2D verticalmente é equivalente a adicionar linhas a um array.

O código a seguir mostra isso.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
row = np.array([7, 8, 9])
arr = np.vstack([arr, row])
print(arr)

Resultado:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Use a função numpy.append() para adicionar uma linha a um array em NumPy

A função append() do módulo numpy pode adicionar elementos ao final do array. Especificando o axis como 0, podemos usar esta função para adicionar linhas a um array.

Por exemplo,

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
row = np.array([7, 8, 9])
arr = np.append(arr, [row], axis=0)
print(arr)

Resultado:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Use a função numpy.r_() para adicionar uma linha a um array em NumPy

A função r_() do módulo numpy concatena arrays combinando-os verticalmente.

Verifique o código abaixo para ver como podemos usar isso para adicionar linhas a um array.

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
row = np.array([7, 8, 9])
arr = np.r_[arr, [row]]
print(arr)

Resultado:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Alternativamente, podemos usar a função concatenate() também. A função concatenate() combina dois ou mais arrays para que possa ser usada para obter o resultado desejado.

Por exemplo,

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
row = np.array([7, 8, 9])
arr = np.concatenate((arr, [row]), axis=0)
print(arr)

Resultado:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

Use a função numpy.insert() para adicionar uma linha a um array em NumPy

A função insert() adiciona objetos ao longo do eixo especificado e da posição. Ele pode ser usado para inserir uma linha em um array em nossa posição específica desejada.

Por exemplo,

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
row = np.array([7, 8, 9])
row_n = arr.shape[0]  # last row
arr = np.insert(arr, row_n, [row], axis=0)
print(arr)

Resultado:

[[1 2 3]
 [4 5 6]
 [7 8 9]]

No código acima, adicionamos a linha no final do array. A função shape() retorna as dimensões do array, que revela o número total de linhas na matriz.

Manav Narula avatar Manav Narula avatar

Manav is a IT Professional who has a lot of experience as a core developer in many live projects. He is an avid learner who enjoys learning new things and sharing his findings whenever possible.

LinkedIn