Jak zmapować funkcję na tablicę numpy (z przykładami)

Aby odwzorować funkcję na tablicę NumPy, możesz użyć następującej podstawowej składni:

 #define function
my_function = lambda x: x*5

#map function to every element in NumPy array
my_function(my_array)

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

Przykład 1: Mapowanie funkcji na jednowymiarową tablicę NumPy

Poniższy kod pokazuje, jak zmapować funkcję na tablicę NumPy, która mnoży każdą wartość przez 2, a następnie dodaje 5:

 import numpy as np

#create NumPy array
data = np. array ([1, 3, 4, 4, 7, 8, 13, 15])

#define function
my_function = lambda x: x*2+5

#apply function to NumPy array
my_function(data)

array([ 7, 11, 13, 13, 19, 21, 31, 35])

Oto sposób obliczenia każdej wartości w nowej tabeli:

• Pierwsza wartość: 1*2+5 = 7
• Druga wartość: 3*2+5 = 11
• Trzecia wartość: 4*2+5 = 13

I tak dalej.

Przykład 2: Mapowanie funkcji na wielowymiarową tablicę NumPy

Poniższy kod pokazuje, jak odwzorować funkcję na wielowymiarową tablicę NumPy, która mnoży każdą wartość przez 2, a następnie dodaje 5:

 import numpy as np

#create NumPy array
data = np. array ([[1, 2, 3, 4], [5, 6, 7, 8]])

#view NumPy array
print (data)

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

#define function
my_function = lambda x: x*2+5

#apply function to NumPy array
my_function(data)

array([[ 7, 9, 11, 13],
       [15, 17, 19, 21]])

Należy zauważyć, że ta składnia działa równie dobrze w przypadku tablicy wielowymiarowej, jak i jednowymiarowej.

Dodatkowe zasoby

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

Jak dodać kolumnę do tablicy NumPy
Jak przekonwertować tablicę NumPy na listę w Pythonie
Jak wyeksportować tablicę NumPy do pliku CSV