Discussion :

[parrax]

Bonjour,
J'ai crée un petit script mais je souhaiterais savoir si je peux (et alors comment) le compacter:

Code :

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

1
2
3
4
5
6
7
8
def matriceVA(n):
  A=[]
  for i in range(n):
    L=[]
    for k in range(n):
      L.append(abs(i-k))
    A.append(L)
  return(A)

Merci d'avance pour votre aide!

[BufferBob]

salut,

via des list comprehensions on peut résumer facilement la fonction :

Code :

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

1
2
def matriceVA(n):
   return [[abs(i-k) for k in range(n)] for i in range(n)]

et si on veut pousser encore un peu le bouchon on peut même simplement se contenter d'un lambda :

Code :

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

1
2
3
>>> matriceVA = lambda n: [[abs(i-k) for k in range(n)] for i in range(n)]
>>> matriceVA(5)
[[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]]

[parrax]

Merci beaucoup

[fifafou]

Pour les liste à plusieurs dimension,je pense que Numpy est mieux,créer une matrice case par case n'est pas très optimisé.
Après c'est juste pour la performance.