【python高速化】list処理の高速化方法を比較してみた

# coding: UTF-8

# 必要なライブラリのインポート
import ramdom

# データの作成（100万人のテストの点数が入ったリストを作成）
test_score_list = [random.randint(0, 100) for i in range(1000000)]

print(test_score_list)
# [95, 87, 31, 99, 75, 35, 38, 36, 91, 31, 23, 36, ・・・・]

print(len(test_score_list))
# 1000000

# forを使った処理

judge_list = []  # True, Falseを格納するリストを作成

for test_score in test_score_list:
    if test_score >= 50:  # 50点以上の場合
        judge_list.append(True)
    else:  　　　　　　　　　# 50点未満の場合
        judge_list.append(False)

print(judge_list)
# [True, True, False, True, True, False・・・]

print(len(judge_list))  # 50点以上の人の合計
# 504953

# 計測時間
78.6 ms ± 2.01 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)

# 内包表記を使った処理

# True, Falseを格納するリストを作成
over_fifty_list = [True if test_score >= 50 else False for test_score in test_score_list]

print(judge_list)
# [True, True, False, True, True, False・・・]

print(len(judge_list))  # 50点以上の人の合計
# 504953

# 計測時間
49.3 ms ± 494 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)

# 内包表記(簡略化ver.)を使った処理

# True, Falseを格納するリストを作成
over_fifty_list = [test_score >= 50 for test_score in test_score_list]

print(judge_list)
# [True, True, False, True, True, False・・・]

print(len(judge_list))  # 50点以上の人の合計
# 504953

# 計測時間
41.6 ms ± 286 µs per loop (mean ± std. dev. of 7 runs, 10 loops each)

# ジェネレーター式を使った処理

# True, Falseを格納するリストを作成
judge_generator = (test_score >= 50 for test_score in test_score_list)


print(judge_generator)
# <generator object <genexpr> at 0x10c849ba0>

print(len(judge_generator))  # 50点以上の人の合計
# TypeError: object of type 'generator' has no len()

print(sum(judge_generator))
# 504953

# 計測時間
320 ns ± 10.9 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

# ユニバーサル関数を使った処理

# 自作関数を作成
def judge_func(test_score):
    """
    50点以上か判定する関数
    """
    if test_score >= 50:
        return True
    else:
        return False

# 自作関数をユニバーサル化する
judge_func_uni = np.frompyfunc(judge_func, 1, 1)  # ユニバーサル関数化

# True, Falseを格納するリストを作成
judge_list = [judge_func_uni(test_score) for test_score in test_score_list]

print(judge_list)
# [True, True, False, True, True, False・・・]

print(len(judge_list))  # 50点以上の人の合計
# 504953

# 計測時間
1.32 s ± 11.1 ms per loop (mean ± std. dev. of 7 runs, 1 loop each)

# map関数を使った処理

# True, Falseを格納するリストを作成
judge_list = list(map(lambda x: True if x >= 50 else False, test_score_list))

print(judge_list)
# [True, True, False, True, True, False・・・]

print(len(judge_list))  # 50点以上の人の合計
# 504953

# 計測時間
81.3 ms ± 1.4 ms per loop (mean ± std. dev. of 7 runs, 10 loops each)