Écrire des boucles plus efficaces

Écrire du code Python efficace

Logan Thomas

Scientific Software Technical Trainer, Enthought

Avertissement concernant la leçon

Certaines des boucles suivantes peuvent être éliminées à l'aide des techniques abordées dans les leçons précédentes.
Les exemples présentés dans cette leçon sont fournis à titre indicatif.

alt=”Warning sign with text Warning: for demonstration purposes only”

Écrire des boucles plus efficaces

Comprendre ce qui est effectué à chaque itération de la boucle
Déplacer les calculs uniques à l'extérieur (au-dessus) de la boucle
Utiliser des conversions holistiques en dehors (en dessous) de la boucle
Toute action effectuée une seule fois devrait être exclue de la boucle

Déplacement des 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])

# Calculate total average once (outside the loop)
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éplacement des 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éplacement des calculs au-dessus d'une boucle

%%timeit
# Calculate total average once (outside the loop)
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)

Utilisation des conversions holistiques

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

Utilisation des conversions holistiques

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.

Écrire du code Python efficace