Pythonでforループを抜けたりスキップする方法

Pythonでforループを特定の条件で止めたいんだけど、どうしたらいいんだっけ・・・

プログラミング言語の中でも比較的覚えやすい Python。

他の言語と同じようにPythonでも繰り返し（ループ）処理が定義できる。

ループは無限ループでなければ指定の回数、もしくはリストを一通り処理すれば終了する。

しかしループ中に目的を満たした場合はループを終了したいし、処理対象にしたくないデータがあった場合はループ中の処理は行わずに次のデータを処理したくなる。

ではそのような場合どうすればいいのか。

そこで今回はPythonでforループを抜けたりスキップする方法について紹介する。

この記事を書いている人

システムエンジニア、AIエンジニアと、IT業界で10年以上働いている中堅。PythonとSQLが得意。

記事を読むメリット

Pythonでforループを抜けたりスキップする方法がわかり、効率的に短い時間でプログラムを動作させられるようになる。

Pythonのforループ

まずはPythonのforループを書いてみる。

# normal loop
for i in range(0,3):
    print("i:{}".format(i))
    for j in range(0, 2):
        print("  j:{}".format(j))

# i:0
#   j:0
#   j:1
# i:1
#   j:0
#   j:1
# i:2
#   j:0
#   j:1

今回は2重ループにしている。

他の書き方もあるので公式ドキュメントを参考にすると色々なループを書くことができる。

参考

4. その他の制御フローツール — Python 3.9.4 ドキュメント

で、ループを止めたりスキップするにはどうしたらいいわけ？

それは次に話します

Pythonでforループを抜ける方法

Pythonでforループを抜ける方法、それは break を使えばできる。

break を使うと以下のようなコードになる。

# break loop1
for i in range(0,3):
    print("i:{}".format(i))
    if i == 1:
        print("break")
        break
    for j in range(0, 2):
        print("  j:{}".format(j))

# i:0
#   j:0
#   j:1
# i:1
# break

先程のコードでは i=2 までループが実行されていたが、ここでは i=1 の段階で break を発生させてループを終了させている。

2重ループでbreakしたらどこが止まるの？

じゃあやってみますか

では2重ループで break したらどうなるのか。

実際にやってみると以下のような結果になる。

# break loop2
for i in range(0,3):
    print("i:{}".format(i))
    for j in range(0, 2):
        print("  j:{}".format(j))
        if j == 0:
            print("break")
            break

# i:0
#   j:0
# break
# i:1
#   j:0
# break
# i:2
#   j:0
# break

今度はiのループではなく、jのループに break を仕掛けたところ、 j=0 でjのループが終了し、親であるiのループは継続しているのがわかる。

なので break で抜けるのは直近のループのみということがわかる。

Pythonでforループのイテレーションをスキップする方法

ではPythonでforループのイテレーションをスキップするにはどうしたらいいのか。

それは continue を使えば解決できる。

# skip loop
for i in range(0,3):
    print("i:{}".format(i))
    if i == 1:
        print("continue")
        continue
    for j in range(0, 2):
        print("  j:{}".format(j))

# i:0
#   j:0
#   j:1
# i:1
# continue
# i:2
#   j:0
#   j:1

このように continue を使うと、 i=1 のイテレーションでの処理はスキップされたが、ループ自体は解除されずに i=2 のイテレーションが実行される。

このように continue を使うと、特定の条件のみループをスキップさせることができる。