Python : probème numpy arange

[Biname]

Salut,
Nouveau en Python, je n'arrive pas a implémenter le if W<0 : W = 0

import matplotlib.pyplot as plt
import numpy as np
#
def T_cos(jj):
    W = (np.cos(np.pi* jj/(365/2)))
    # if W < 0 : W = 0    # comment implémenter ceci ?
    return (W)
#
j = np.arange(1, 366, 1)
plt.plot(T_cos(j))
plt.show()
#

Ou, compilateur en ligne : http://tpcg.io/_L4NWFA

Biname
-----

[umfred]

fait un print(W) tu verras que W n'est un seul nombre mais un np.array 1D
il faut utiliser np.where
https://numpy.org/doc/stable/referen...mpy.where.html

import matplotlib.pyplot as plt
import numpy as np
#
def T_cos(jj):
    W = (np.cos(np.pi* jj/(365/2)))
    W=np.where(W<0,0,W)
    return (W)
#
j = np.arange(1, 366, 1)
plt.plot(T_cos(j))
plt.show()
#

[pm42]

Et ne pas utiliser des compilos en ligne simplistes mais des IDE notamment les gratuits permet de faire facilement du pas à pas, d'examiner la valeur des variables, de taper pendant un breakpoint des expressions...

[Biname]

Salut,

fait un print(W) tu verras que W n'est un seul nombre mais un np.array 1D
il faut utiliser np.where
https://numpy.org/doc/stable/referen...mpy.where.html

Merci
Oui pour l'array-1, les erreurs renvoyées par l'interpréteur étaient claires !
Pas pensé à chercher dans numpy !
J'avais trouvé ceci, un peu lourd :

import matplotlib.pyplot as plt
import numpy as np
#
j = np.arange(1, 366, 1)
W = j + 0.1           # cree array w(0 à 364), + 0.1 force float 
print("W[0] = " + str(W[0]))
for x in range (0,365):
    a = np.cos(np.pi* x/(365/2))
    if a > 0:        a = 0
    W[x] = a
#
plt.plot(W)
plt.show()

Avec numpy, il faut utiliser numpy !

Biname

[A voir en vidéo sur Futura]