fonctions sur numpy.array

Bonjour à tous,

je développe un script pour faire du calcul sur des directions et vitesses de vents; et j'ai besoin de définir au préalable une fonction de kernel gaussien et d'epanechnikov.

Mon problème vient de la deuxième. Voici comment je l'ai définie:

Code:

---

1 2 3 4 5

def epanechK(x): y=0 if -1<x<1: y=0.75*(1-x**2) return y

---

dans la pratique, x = (X-wd[i])/sig ; avec X un array, wd[i] un élément d'une liste, et sig une constante.

et quand je fais tourner le script, j'ai cette erreur:
"ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()"

j'ai donc changé la définition de la fonction par:

Code:

---

1 2 3 4 5

def epanechK(x): y=0 if all([x<1,x>-1]): y=0.75*(1-x**2) return y

---

Mais rien n'y fait, j'ai toujours ce même message d'erreur.

Quelqu'un connait-il la bonne syntaxe a utiliser??


P.S.: pour info, le script marche très bien lorsque je ne met aucune condition sur x....

Salut,

Citation:

---

Envoyé par jeps76

"ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()"

---

Le message d'erreur dit que all est une méthode de l'objet a, or tu l'utilises comme une fonction. si X est ton objet numpy, il faudrait écrire X.all()

A+

Pfeuh

D'accord... mais du coup, comment faire passer la condition?
x.all(x>-1, x<1)??

Citation:

---

Envoyé par jeps76

D'accord... mais du coup, comment faire passer la condition?
x.all(x>-1, x<1)??

---

Je ne sais pas ce que tu veux faire, mais peut-être comme ça?

Code:

---

1 2	if X.all() > -1 and X.all() < 1: # traitement

---

Yes, ça marche sans soucis!! merci beaucoup ;)

Code:

---

1 2 3 4 5

def epanechK(x): y=0 if -1<x<1: y=0.75*(1-x**2) return y

---

C'est mieux avec where:

Code:

---

1 2 3 4

import numpy as np def epanechK(x): return np.where(np.and(x<1, -1<x), 0.75*(1-x**2), 0)

---

Ou quelque chose du style.