plusieurs lignes similaires en python

Question

Ma question concerne le style de codage: Quand je veux appliquer une procédure à plusieurs variables, comme:

from pylab import vstack  # the function to apply to many variables 
v1 = range(5)  # some dummy variables 
v2 = range(9) 
... 
v20 = range(8) 

# now, I want to apply a function to all variables that changes them! 
v1 = vstack(v1) # I really do not need the "old" v1 anymore 
v2 = vstack(v2) 
... 
v20 = vstack(v20) 

Ce sont 20 lignes de presque le même code. Existe-t-il un moyen d'écrire les conversions (vx = vstack(vx)) en moins de lignes?

Les mettre dans un itérables ne fonctionne pas:

all_v = [v1, v2, v3, .... , v20] 
for v in all_v: 
    v = vstack(v) 

type (v1) # -> v1: <type: list> 

Answer 1

Avez-vous vraiment besoin d'être en variables distinctes pour commencer? C'est généralement une odeur de code. Je les garde dans une liste ou un dict si les noms sont importants, et mise à jour:

my_data = {} 
my_data['v1'] = range(5) 
my_data['v2'] = range(9) 
...etc... 

my_data = {k: vstack(v) for k, v in my_data.iteritems()} 

Answer 2

from pylab import vstack 
parameters = (5, 9, ..., 8) 

v = [vstack(range(p)) for p in parameters] 

Cela exécutera une compréhension de la liste de tuple parameters, qui contient vos paramètres à la range fonction. Pour chaque paramètre de ce tuple, le résultat de la fonction range est transmis à la fonction vstack. Les résultats sont rassemblés dans une liste.

Ainsi v[0] correspond au résultat de la valeur du paramètre parameters[0] etc.

Dans votre code, il ne fonctionne pas parce que vous écrasez localement la variable v dans la boucle. Cela ne changera pas l'élément dans la liste d'origine. Si vous voulez créer un (nouveau) liste de ces résultats, vous pouvez utiliser list.append:

all_v = [v1, v2, v3, .... , v20] 
new_v = [] 
for v in all_v: 
    new_v.append(vstack(v))