“Numpy Appliquez une fonction à la table” Réponses codées

Numpy Appliquez une fonction à la table

import numpy as np

arr = np.array([1,2,3,4])
print(np.apply_along_axis(lambda x : x ** 2, 0, arr))

#Output: array([ 1,  4,  9, 16])