Jak dodać wiersz do macierzy w numpy (z przykładami)

Aby dodać wiersz do macierzy w NumPy, możesz użyć następującej składni:

 #add new_row to current_matrix
current_matrix = np. vstack ([current_matrix, new_row])

Możesz także użyć poniższej składni, aby dodać tylko wiersze do macierzy, która spełnia określony warunek:

 #only add rows where first element is less than 10
current_matrix = np. vstack ((current_matrix, new_rows[new_rows[:,0] < 10 ]))

Poniższe przykłady pokazują, jak używać tej składni w praktyce.

Przykład 1: Dodaj wiersz do macierzy w NumPy

Poniższy kod pokazuje, jak dodać nowy wiersz do macierzy w NumPy:

 import numpy as np

#define matrix
current_matrix = np. array ([[1,2,3], [4, 5, 6], [7, 8, 9]])

#define row to add
new_row = np. array ([10, 11, 12])

#add new row to matrix
current_matrix = np. vstack ([current_matrix, new_row])

#view updated matrix
current_matrix

array([[ 1, 2, 3],
       [4,5,6],
       [7, 8, 9],
       [10, 11, 12]])

Zauważ, że ostatni wiersz został pomyślnie dodany do macierzy.

Przykład 2: Dodaj wiersze do macierzy na podstawie warunku

Poniższy kod pokazuje, jak dodać wiele nowych wierszy do istniejącej macierzy na podstawie określonego warunku:

 import numpy as np

#define matrix
current_matrix = np. array ([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

#define potential new rows to add
new_rows = np. array ([[6, 8, 10], [8, 10, 12], [10, 12, 14]])

#only add rows where first element in row is less than 10
current_matrix = np. vstack ((current_matrix, new_rows[new_rows[:,0] < 10 ]))

#view updated matrix
current_matrix

array([[ 1, 2, 3],
       [4,5,6],
       [7, 8, 9],
       [6, 8, 10],
       [8, 10, 12]])

Dodano tylko wiersze, których pierwszy element był mniejszy niż 10.

Uwaga : pełną dokumentację online dotyczącą funkcji vstack() można znaleźć tutaj .

Dodatkowe zasoby

Poniższe samouczki wyjaśniają, jak wykonywać inne typowe operacje w NumPy:

Jak znaleźć indeks wartości w tablicy NumPy
Jak dodać tablicę Numpy do Pandas DataFrame
Jak przekonwertować tablicę NumPy na Pandas DataFrame