namedtuple
標準モジュール collections または typing 内にある機能。c++でいう構造体のような機能である。
「複数の値をまとめて持てる」「イミュータブル」というTupleの性質を持ちつつ、「属性に名前を付けられる」という機能が追加されている。
これにより、Listなどのようにどこかで勝手に変更されないという安心感を持ちつつ、可読性を高めて要素にアクセスできる。 しかもドットアクセスなので、IDEが対応していれば補完が効く。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# 標準Tupleの生成と要素アクセスtup1 = ('pi', 3.14, [0, 1], min)print(tup1[0]) # => piprint(tup1[1]) # => 3.14# NamedTupleの生成と要素アクセスfrom collections import namedtuple# まず(クラス名, 各カラム名)を渡してクラスを定義するTestTuple = namedtuple('TestTuple', 'name value list, func')# それを使って生成するtup2 = TestTuple('pi', 3.14, [0, 1], min)print(tup2.name) # => piprint(tup2.value) # => 3.14 |
実態はクラス
ちなみに実態はクラスなので、速度が向上するとかは望めない（むしろdict等と比べて遅い）。
1 2 |
# ...print(type(tup2)) # => <class '__main__.TestTuple'> |
つまり、namedtuple自体はクラスファクトリーである。
namedtupleという名称も、むしろTupleより「クラスに、属性を勝手に変更できない機能を追加したもの」といった方が近い気がする。
Pickle化の注意点
Pickle化する際には他のユーザ定義クラスと同様の手間がかかる。つまり、
- 保存時はグローバルスコープに定義されていないといけない
- 保存されるのはクラス名と属性名のみで、クラス構造自体は保存されない
- unpickle時、そのスコープから同名のクラスが見えないといけない
- しかし、同名であれば、クラス構造が異なっても読み込めてしまう
書き出すとき、グローバルスコープからnamedtupleインスタンスが見えてないとエラー
1 2 3 4 5 6 7 8 9 10 11 12 |
def f(): Test = namedtuple('Test', 'name value') tup = Test('pi', 3.14) return tuptup2 = f() # 関数内で定義したNamedTupleインスタンスを返値で得るprint(tup2.name) # その場で使うことはできるwith open('some.pickle', 'wb') as wh: # pickleするとエラー pickle.dump(tup2, wh)# => _pickle.PicklingError: Can't pickle <class '__main__.Test'>: attribute lookup Test on __main__ failed |
グローバルでNamedTupleを定義するとOK（保存は関数内・外どちらでも）
1 2 3 4 5 6 7 8 9 |
Test = namedtuple('Test', 'name value')def f(): tup = Test('pi', 3.14) with open('some.pickle', 'wb') as wh: pickle.dump(tup, wh)f() |
読み込むとき、そのスコープから同名のクラスが見えないとエラー
何も情報を与えないと、Testクラスが見えないと言われる。
1 2 3 4 |
with open('some.pickle', 'rb') as rh: tup = pickle.load(rh)# => AttributeError: Can't get attribute 'Test' on <module '__main__' from '(実行したpythonスクリプトパス)'> |
importする必要がある。明示的にはスクリプト内で使われてないので、IDEの設定によっては最適化で自動で除かれてしまう点に注意。
1 2 3 4 |
from (保存したpythonスクリプト) import Testwith open('some.pickle', 'rb') as rh: tup = pickle.load(rh) |
そのスクリプト内で同じものを定義しても通る。
しかし、あまり同じ定義が複数ファイルに分かれるのは良くない上に、属性名は違うものを定義しても通ってしまうので、使わない方がよい。
1 2 3 4 5 6 7 8 9 10 11 12 |
Test = namedtuple('Test', 'name2 value2') # 属性名を変えたTestを定義with open('some.pickle', 'rb') as rh: tup = pickle.load(rh) # 読み込めはするprint(tup.name) # => piprint(tup.name2) # => AttributeError: 'Test' object has no attribute 'name2'# スクリプト内で新たに生成したTestと属性名が食い違ってしまうtup2 = Test('r2', 1.414)print(tup2.name2) # => r2print(tup2.name) # => AttributeError: 'Test' object has no attribute 'name' |
さらに、namedtupleでもないクラスでも通ってしまう。
1 2 3 4 5 6 |
class Test: def greet(self): print('Hello!')with open('some.pickle', 'rb') as rh: tup = pickle.load(rh) # 読み込めてしまう |
JSON化
Pickle化あたりの挙動が困る場合、容量や、変換時の処理時間は多少かかるが、dictを仲介してJSONなどで保存するのも手。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import jsonTest = namedtuple('Test', 'name value')tup = Test('pi', 3.14)# NamedTuple -> JSONtup_json = json.dumps(tup._asdict())print(tup_json)# => {"name": "pi", "value": 3.14}# JSON -> NamedTupletup2 = Test(**json.loads(tup_json))print(tup2)# => Test(name='pi', value=3.14) |
型指定
typing.NamedTuple で定義すれば、型も指定できる。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from typing import NamedTuple# collections.namedtupleと同じ感じで指定する方法Test1 = NamedTuple('Test1', [('name', str), ('value', float)])tup1 = Test1('pi', 3.14)print(tup1)# classを継承して指定する方法class Test2(NamedTuple): name: str value: floattup2 = Test2('pi', 3.14)print(tup2) |
DataClass
Python3.7以降では、DataClassデコレータが新しく使えるようになり、ほぼ同等の機能が実現できる。
一部だけミュータブルにしたり、インスタンスを直接比較する際に使う属性を決めたり、細かな設定も可能になっているため、こちらの方が使いやすいかも。 （どちらも糖衣構文なだけで、できあがるものは、いろいろ設定が付与されたクラスである点に違いは無い）
速度は、単に生成するだけならNamedTupleが速い（といっても誤差の範囲）が、アクセスなどその他諸々の速度はDataClassが速い、という記事あり。
展開代入はNamedTupleにできてDataClassにできないらしいので、それを使いたい場合はNamedTupleを採用する価値があるが、基本、DataClassに取って代わられそう。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# NamedTupleTest1 = namedtuple('Test', 'name value')tup1 = Test1('pi', 3.14)a, b = tup1 # OK# DataClass@dataclassclass Test2: name: str value: floattup2 = Test2('pi', 3.14)a, b = tup2 # TypeError: cannot unpack non-iterable Test2 objecta, b = vars(tup2).values() # OK |
NamedTupleを使っていたところをDataClassに置き替えたとしても、 「生成」と「要素アクセス」の基本的な機能しか使っていないのであれば両者の書き方は変わらないので、定義さえ書き換えればよい。
