Matrices conditionnelles avec numpy/scipy
Bonjour,
Je souhaite remplir une matrice selon des conditions portant sur les éléments d'autres matrices.
Soit les matrices de départ (entre parenthèses les dimensions) :
S (4x1)
Y (4x4)
P (4x4)
Le but est de construire la matrice T (4x9).
Pour l'exemple, je prends T(i,1), T(i,2), Y(i,1) (les autres colonnes se construisent de la même manière).
Les conditions sont :
Si S(i,1)>0Si Y(i,1)<0
T(i,1) = -S(i,1) x Y(i,1) + P(i,1) + Y(i,1)xPp
Si P(i,2) > 0
T(i,2) = -S(i,1) x Y(i,1)
Sinon
T(i,2) = 10^7
Sinon
T(i,1) = S(i,1) - Y(i,1) + P(i,1) - Pp
Si P(i,2) > 0
T(i,2) = S(i,1) - Y(i,1)
Sinon
T(i,2) = 10^7
Avec :
Si P(i,4) <= S(i,1)
Pp = P(i,3)
Sinon
Pp = 0
Plutôt que de retranscrire les matrices en tableau pour accéder aux élément et faire les comparaisons, j'ai vu qu'il existe la possibilité avec numpy de faire les tests directement en créant une matrice. Mais les tests portent sur des valeurs fixées et non sur des valeurs prises dans d'autre matrices.
Existe-t-il une fonction que je n'aurais pas encore trouvée ?
