Numpy - Modifier valeurs sous condition

Bonjour,

J'ai un tableau numpy, disons par exemple:

Code:

---

a=np.array([[0,1,2,4],[4,3,6,8],[9,5,7,2],[8,9,8,5],[0,5,3,8]])

---

Je veux rajouter une 5e colonne qui ait pour valeurs soit la col 1 si celle-ci est inférieure à la col 0, soit la colonne 0 si la condition n'est pas respectée.
Avec des booléens, si la condition est respectée, on a :

Code:

---

x[:,1] < x[:,0]

---

qui retourne

Code:

---

array([False, True, False, True], dtype=bool)

---

e sais insérer une 5e col :

Code:

---

np.hstack((a,np.zeros((5,1))))

---

Question: comment faire pour remplacer les valeurs de la 5e col pour parvenir à :

Code:

---

1 2 3 4 5

array([[0, 1, 2, 4, 0], [4, 3, 6, 8, 3], [9, 5, 7, 2, 5], [8, 9, 8, 5, 8], [0, 5, 3, 8, 0]])

---

De préférence en évitant les boucles (comme

Code:

---

1 2	for i in range(5): a[i][4]=a[i][1] if a[i][1]<a[i][0] else a[i][0]

---

D'avance, merci
Dominique

Salut,
Avec where par exemple :

Code:

---

1 2 3 4 5 6 7 8 9

>>> a = np.array([[0,1,2,4],[4,3,6,8],[9,5,7,2],[8,9,8,5],[0,5,3,8]]) >>> col = np.where(a[:, 1] < a[:, 0], a[:, 1], 0).reshape(-1, 1) >>> np.hstack((a, col)) array([[0, 1, 2, 4, 0], [4, 3, 6, 8, 3], [9, 5, 7, 2, 5], [8, 9, 8, 5, 0], [0, 5, 3, 8, 0]]) >>>

---

Salut Dardanos,

Merci beaucoup !!!
C'est exactement ce que je voulais.
Je vais enfin un peu mieux maîtriser Numpy, qui gagne à être connu.

Dominique

J'avais mal lu la consigne. Il y a plus simple avec :

Code:

---

col = np.minimum(a[:, 0], a[:, 1]).reshape(-1, 1)

---

Pas grave !
Ce que je voulais, c'est comprendre mieux le principe ...
Merci beaucoup