Pythonで辞書を特定の長さに切り詰める方法（itertools.islice活用）

Pythonの辞書を指定した件数で切り詰める方法

Pythonの標準ライブラリに含まれる itertools モジュールには、ClojureやHaskellといった関数型言語から着想を得た、便利なイテレータ構成ブロックが数多く用意されています。その中のひとつが islice() 関数です。

islice() は、イテラブル（反復可能なオブジェクト）から特定の範囲の要素を選び出し、新しいイテレータとして返す関数です。スライス操作をイテレータに対して行えるため、大きなデータを扱う際にもメモリ効率が良いという特徴があります。

islice() の基本構文

islice(sequence, start, stop, step)

• sequence：対象となるイテラブルオブジェクト
• start：抽出を開始する位置（省略可能）
• stop：抽出を終了する位置
• step：要素を取得する間隔（省略可能）

辞書を最初の3項目に切り詰める例

次の例では、5つの要素を持つ辞書から islice() を使って最初の3項目のみを選択し、新しい辞書を作成しています。辞書の items() メソッドでキーと値のペアを取得し、それを islice() に渡している点がポイントです。

>>> D1={"pen":25, "pencil":10, "book":100, "sharpner":5, "eraser":5}
>>> import itertools
>>> D2=dict(itertools.islice(D1.items(),3))
>>> D2
{'pen': 25, 'pencil': 10, 'book': 100}

このように、dict(itertools.islice(D.items(), n)) と書くだけで、元の辞書の先頭からn個の要素を持つ新しい辞書を簡単に作成できます。元の辞書 D1 は変更されないため、安全に処理を行えます。

なお、Python 3.7以降では辞書が挿入順序を保持するため、この方法で「先頭n件」を確実に取り出せます。それ以前のバージョンでは順序が保証されないため注意が必要です。