Discussion :

[hazem2410]

Bonjour,

J'ai un np.array A, tel que:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
 
A = np.array([0,1,2,3,4])

Je veux avoir la matrice M équivalente, qui a les éléments suivants: M[ij] = |A[i] - A[j]|. Ce qui donnera à partir de mon exemple ceci:

array([[0, 1, 2, 3, 4],
[1, 0, 1, 2, 3],
[2, 1, 0, 1, 2],
[3, 2, 1, 0, 1],
[4, 3, 2, 1, 0]])

Evidemment une boucle for peut faire l'affaire. Toutefois, je travaille sur de très grandes données avec des millions de simulations, alors j'évite au maximum les boucles for qui sont lourdes sur Python. Quelqu'un a une idée sur comment obtenir ce résultat?

[__dardanos__]

Salut,
Utilise la méthode tile pour vectoriser les opérations.

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
>>> import numpy as np
>>> A = np.array([0,1,2,3,4])
>>> np.abs(np.tile(A, (5, 1)) - np.tile(A, (5, 1)).T)
array([[0, 1, 2, 3, 4],
       [1, 0, 1, 2, 3],
       [2, 1, 0, 1, 2],
       [3, 2, 1, 0, 1],
       [4, 3, 2, 1, 0]])
>>>

[hazem2410]

Merci bien