把迴圈寫得更好

撰寫高效的 Python 程式碼

Logan Thomas

Scientific Software Technical Trainer, Enthought

課程注意事項

下面有些迴圈可用前面課程的技巧來省略。
本課範例僅供「示範」用途。

alt=「警告標誌，內有文字：Warning: for demonstration purposes only」

把迴圈寫得更好

了解每次迭代在做什麼
一次性計算移到迴圈外（上方）
整體轉換放在迴圈外（下方）
凡是只做「一次」的，都應在迴圈外

把計算移到迴圈上方

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!

把計算移到迴圈上方

%%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)

把計算移到迴圈上方

%%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)

使用整體轉換

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

使用整體轉換

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)

一起來練習吧！

撰寫高效的 Python 程式碼