Bibliothèque mathématique et tableaux en Python

Question

Je dois utiliser la bibliothèque mathématique pour effectuer des calculs sur un tableau.
J'ai essayé quelque chose comme ceci:

import numpy as np 
import math 
a = np.array([0, 1, 2, 3]) 
a1 = np.vectorize(a) 
print("sin(a) = \n", math.sin(a1)) 

Malheureusement, il ne fonctionne pas. Une erreur s'est produite: "TypeError: must be real number, not vectorize". Comment puis-je utiliser la fonction vectorize pour calculer ce genre de choses?

Answer 1

import numpy as np 
import math 
a = np.array([0, 1, 2, 3]) 
print("sin(a) = \n", [math.sin(x) for x in a]) 

math.sin nécessite un nombre réel à la fois.

Answer 2

Le point entier de numpy est que vous ne avez pas besoin de méthode math ou toute compréhension de la liste:

>>> import numpy as np 
>>> a = np.array([0, 1, 2, 3]) 
>>> a + 1 
array([1, 2, 3, 4]) 
>>> np.sin(a) 
array([ 0.  , 0.84147098, 0.90929743, 0.14112001]) 
>>> a ** 2 
array([0, 1, 4, 9]) 
>>> np.exp(a) 
array([ 1.  , 2.71828183, 7.3890561 , 20.08553692]) 

Vous pouvez utiliser a comme si elle était un scalaire et vous obtenez le tableau correspondant.

Si vous avez vraiment besoin d'utiliser math.sin (indice: vous ne), vous pouvez vectorize elle (la fonction elle-même, pas le tableau):

>>> vsin = np.vectorize(math.sin) 
>>> vsin(a) 
array([ 0.  , 0.84147098, 0.90929743, 0.14112001])