Як зіставити функцію з масивом numpy (з прикладами)

Щоб зіставити функцію з масивом NumPy, можна використовувати наступний базовий синтаксис:

 #define function
my_function = lambda x: x*5

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

Наступні приклади показують, як використовувати цей синтаксис на практиці.

Приклад 1: функція зіставлення з одновимірним масивом NumPy

Наступний код показує, як зіставити функцію з масивом NumPy, який множить кожне значення на 2, а потім додає 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])

Ось як було обчислено кожне значення в новій таблиці:

• Перше значення: 1*2+5 = 7
• Друге значення: 3*2+5 = 11
• Третє значення: 4*2+5 = 13

І так далі.

Приклад 2: функція зіставлення з багатовимірним масивом NumPy

У наведеному нижче коді показано, як зіставити функцію з багатовимірним масивом NumPy, який множить кожне значення на 2, а потім додає 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]])

Зауважте, що цей синтаксис так само добре працює як з багатовимірним, так і з одновимірним масивом.

Додаткові ресурси

У наступних посібниках пояснюється, як виконувати інші типові операції в NumPy:

Як додати стовпець до масиву NumPy
Як перетворити масив NumPy на список у Python
Як експортувати масив NumPy у файл CSV