Pythonで複数のリストを同時に反復処理する方法：zip()関数の使い方

Pythonで複数のリストを同時に反復処理したい場合には、zip()関数が便利です。zip()は任意の個数（n個）のイテラブルを受け取り、タプルのリストを返します。返されるリストのi番目のタプルは、それぞれのイテラブルのi番目の要素を組み合わせて構成されます。

基本的な使用例

L1 = [1, 2, 3, 4]
L2 = ['aa', 'bb', 'cc', 'dd']
L = list(zip(L1, L2))
print(L)

出力

[(1, 'aa'), (2, 'bb'), (3, 'cc'), (4, 'dd')]

アルゴリズム

Step 1: まず3つのユーザー入力リストを作成する
Step 2: zip()関数を使用する
Step 3: タプルを出力する

サンプルコード

次の例では、学籍番号・名前・年齢という3つのリストをユーザー入力から作成し、zip()関数を使って同時に反復処理しています。

# zip()関数を使って3つのリストを反復処理する
import itertools
A = list()
B = list()
C = list()
n = int(input("何件のデータを保存しますか？"))
print("学籍番号を入力してください")
for i in range(n):
    k = int(input(""))
    A.append(k)
print("名前を入力してください")
for j in range(n):
    k1 = input("")
    B.append(k1)
print("年齢を入力してください")
for j in range(n):
    k1 = int(input(""))
    C.append(k1)
print("\nzip()による反復処理")
for (a, b, c) in zip(A, B, C):
    print(a, b, c)

実行結果

何件のデータを保存しますか？4
学籍番号を入力してください
56
89
67
34
名前を入力してください
yui
jkl
vm
dfg
年齢を入力してください
7
9
6
5
zip()による反復処理
56 yui 7
89 jkl 9
67 vm 6
34 dfg 5

補足：リストの長さが異なる場合の注意点

zip()は、渡されたイテラブルの中で最も短いものに合わせて処理を終了します。そのため、長さの異なるリストを扱うと、余分な要素は無視されます。すべての要素を確実に処理したい場合は、標準ライブラリitertoolsのzip_longest()を使うと、足りない部分を指定した値で埋めながら反復処理できます。