Améliorer vos boucles

Rédiger du code Python efficace

Logan Thomas

Scientific Software Technical Trainer, Enthought

Mise en garde de la leçon

Certaines boucles ci-dessous peuvent être évitées grâce aux techniques vues plus tôt.
Les exemples de cette leçon servent uniquement à des fins démonstratives.

alt="Panneau d'avertissement avec le texte Avertissement : à des fins de démonstration seulement"

Améliorer vos boucles

Comprendre ce que fait chaque itération de boucle
Placer les calculs uniques au-dessus de la boucle
Faire les conversions globales en dehors (après) la boucle
Tout ce qui n'est fait qu'une fois doit être hors de la boucle

Déplacer les calculs au-dessus d'une boucle

import numpy as np

names = ['Absol', 'Aron', 'Jynx', 'Natu', 'Onix']
attacks = np.array([130, 70, 50, 50, 45])

for pokemon,attack in zip(names, attacks):

    total_attack_avg = attacks.mean()

    if attack > total_attack_avg:
        print(
            "{}'s attack: {} > average: {}!"
            .format(pokemon, attack, total_attack_avg)
        )

Absol's attack: 130 > average: 69.0!
Aron's attack: 70 > average: 69.0!

import numpy as np

names = ['Absol', 'Aron', 'Jynx', 'Natu', 'Onix']
attacks = np.array([130, 70, 50, 50, 45])

# Calculer la moyenne totale une fois (à l'extérieur de la boucle)
total_attack_avg = attacks.mean()

for pokemon,attack in zip(names, attacks):

    if attack > total_attack_avg:
        print(
            "{}'s attack: {} > average: {}!"
            .format(pokemon, attack, total_attack_avg)
        )

Absol's attack: 130 > average: 69.0!
Aron's attack: 70 > average: 69.0!

Déplacer les calculs au-dessus d'une boucle

%%timeit
for pokemon,attack in zip(names, attacks):

    total_attack_avg = attacks.mean()

    if attack > total_attack_avg:
        print(
            "{}'s attack: {} > average: {}!"
            .format(pokemon, attack, total_attack_avg)
        )

74.9 µs ± 3.42 µs per loop (mean ± std. dev. of 7 runs, 10000 loops each)

Déplacer les calculs au-dessus d'une boucle

%%timeit
# Calculer la moyenne totale une fois (à l'extérieur de la boucle)
total_attack_avg = attacks.mean()

for pokemon,attack in zip(names, attacks):

    if attack > total_attack_avg:
        print(
            "{}'s attack: {} > average: {}!"
            .format(pokemon, attack, total_attack_avg)
        )

37.5 µs ± 281 ns per loop (mean ± std. dev. of 7 runs, 10000 loops each)

Utiliser des conversions globales

names = ['Pikachu', 'Squirtle', 'Articuno', ...]
legend_status = [False, False, True, ...]
generations = [1, 1, 1, ...]

poke_data = []
for poke_tuple in zip(names, legend_status, generations):

    poke_list = list(poke_tuple)

    poke_data.append(poke_list)

print(poke_data)

[['Pikachu', False, 1], ['Squirtle', False, 1], ['Articuno', True, 1], ...]

Utiliser des conversions globales

names = ['Pikachu', 'Squirtle', 'Articuno', ...]
legend_status = [False, False, True, ...]
generations = [1, 1, 1, ...]

poke_data_tuples = []
for poke_tuple in zip(names, legend_status, generations):
    poke_data_tuples.append(poke_tuple)

poke_data = [*map(list, poke_data_tuples)]

print(poke_data)

[['Pikachu', False, 1], ['Squirtle', False, 1], ['Articuno', True, 1], ...]

%%timeit
poke_data = []
for poke_tuple in zip(names, legend_status, generations):
    poke_list = list(poke_tuple)
    poke_data.append(poke_list)

261 µs ± 23.2 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

%%timeit
poke_data_tuples = []
for poke_tuple in zip(names, legend_status, generations):
    poke_data_tuples.append(poke_tuple)

poke_data = [*map(list, poke_data_tuples)]

224 µs ± 1.67 µs per loop (mean ± std. dev. of 7 runs, 1000 loops each)

Passons à la pratique !

Rédiger du code Python efficace