Python - イテレータ
こんにちは、Pythonプログラマー志望の皆さん!今日は、Pythonのイテレータの世界に魅力的な旅に出かけましょう。あなたの親しみのある近所のコンピューターサイエンスの先生として、私はこの興味深いトピックをガイドすることができて嬉しいです。だから、お気に入りの飲み物を持って、快適に座って、一緒に飛び込もう!
Python イテレータ
イテレータとは?
カラフルなレゴブリックの大きな箱を想像してみてください。イテレータは、その箱に手を入れて一つずつレゴブリックを取り出す魔法の手のようなものです。これにより、全てを床にぶつけずに一つずつブリックを詳細に見つめることができます。Pythonでは、イテレータは同様に動作し、データのコレクションを一つずつ処理することができます。
イテレータはどのように動作するの?
Pythonのイテレータは、二つの特別なメソッド __iter__()
と __next__()
を実装するオブジェクトです。今はよくわからないかもしれませんが、ひとつずつ説明していきましょう!
-
__iter__()
メソッドは、イテレータオブジェクト自体を返します。"さあ、レゴブリックを渡す準備ができたよ!" と言うようなものです。 -
__next__()
メソッドは、シーケンスの次のアイテムを返します。箱から次のレゴブリックを取り出すようなものです。
以下の簡単な例で見てみましょう:
# リストを作成(レゴブリックの箱)
my_list = [1, 2, 3, 4, 5]
# リストからイテレータを取得
my_iterator = iter(my_list)
# next()を使って一つずつアイテムを取得
print(next(my_iterator)) # 出力: 1
print(next(my_iterator)) # 出力: 2
print(next(my_iterator)) # 出力: 3
この例では、 iter(my_list)
がリストのイテレータオブジェクトを作成し、 next(my_iterator)
を毎回呼び出してリストの次のアイテムを取得しています。
イテレータをループで使用するパワー
楽しい事実ですが、Pythonで for
ループを使用すると、背後でイテレータを使用しているのです!どのように動作するのか見てみましょう:
my_list = ["apple", "banana", "cherry"]
for fruit in my_list:
print(f"I love {fruit}!")
# 出力:
# I love apple!
# I love banana!
# I love cherry!
Pythonは自動的に my_list
からイテレータを作成し、ループのたびに __next__()
を使って各アイテムを取得します。素晴らしいでしょうか?
イテレータのエラーハンドリング
さて、魔法の手が空の箱に触れたらどうなるでしょうか?Pythonの言葉で言うと、イテレータにアイテムがもうない場合、何が起こるのでしょうか?これがエラーハンドリングの登場です。
イテレータが消耗された(アイテムがもうない)場合、 StopIteration
例外を発生させます。見てみましょう:
my_list = [1, 2, 3]
my_iterator = iter(my_list)
print(next(my_iterator)) # 出力: 1
print(next(my_iterator)) # 出力: 2
print(next(my_iterator)) # 出力: 3
print(next(my_iterator)) # StopIteration例外が発生
このように優雅に処理するために、try-exceptブロックを使用することができます:
my_list = [1, 2, 3]
my_iterator = iter(my_list)
try:
while True:
item = next(my_iterator)
print(item)
except StopIteration:
print("イテレータの終わりに達しました!")
# 出力:
# 1
# 2
# 3
# イテレータの終わりに達しました!
この方法で、すべてのアイテムを処理し、イテレータの終わりをスムーズに処理することができます。
カスタムイテレータ
イテレータの動作を理解したので、自分のイテレータを作成してみましょう!カウントダウンのイテレータを作りたいと想像してみてください。以下のように作成できます:
class Countdown:
def __init__(self, start):
self.start = start
def __iter__(self):
return self
def __next__(self):
if self.start <= 0:
raise StopIteration
self.start -= 1
return self.start + 1
# カスタムイテレータを使用
countdown = Countdown(5)
for number in countdown:
print(number)
# 出力:
# 5
# 4
# 3
# 2
# 1
この例では、 Countdown
クラスがイテラブル( __iter__()
メソッドを持つ)であり、またイテレータ( __next__()
メソッドを持つ)であるように作られています。 __next__()
が毎回呼ばれるたびに、カウントダウンの次の数字を返します。
非同期イテレータ
さらに高度な領域に進むことにしましょう。非同期プログラミングにおける非同期イテレータについて簡単に触れましょう。これは並列コードを書く方法の一つです。
非同期イテレータは通常のイテレータに似ていますが、 async
と await
キーワードを使用します。以下の簡単な例を見てみましょう:
import asyncio
class AsyncCountdown:
def __init__(self, start):
self.start = start
def __aiter__(self):
return self
async def __anext__(self):
await asyncio.sleep(1) # 非同期操作をシミュレート
if self.start <= 0:
raise StopAsyncIteration
self.start -= 1
return self.start + 1
async def main():
async for number in AsyncCountdown(5):
print(number)
asyncio.run(main())
# 出力(1秒遅延ごと):
# 5
# 4
# 3
# 2
# 1
この非同期イテレータも先ほどの Countdown
クラスと同様に動作しますが、非同期操作(ここでは asyncio.sleep(1)
でシミュレート)を許可します。
イテレータメソッド一覧
以下は、今回説明した主要なメソッドのまとめです:
メソッド | 説明 | 使用される場所 |
---|---|---|
__iter__() |
イテレータオブジェクトを返す | 通常のイテレータ |
__next__() |
シーケンスの次のアイテムを返す | 通常のイテレータ |
__aiter__() |
非同期イテレータオブジェクトを返す | 非同期イテレータ |
__anext__() |
非同期シーケンスの次のアイテムを返す | 非同期イテレータ |
それでは、皆さん!Python イテレータの土地を旅することができましたね。レゴブリックを組み立てることと同じに、イテレータを使いこなすには練習が必要です。だから、自分のイテレータを作って試してみてください。お楽しいコーディングを、そしてイテレータがあなたと共にあることを!
Credits: Image by storyset