compacter un programme (matrices)

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

Code:

---

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!

salut,

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

Code:

---

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:

---

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]]

---

Merci beaucoup :D

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.