Python で2つの配列を for 文で扱いたい場合によく使うのが zip() です。
zip()を使った for 文では暗黙的に同じ大きさが要求されると思っていたが、実際には以下のように2つの配列の大きさが異なっていてもエラーが出ないことに気が付かず、困ったことがあった。
|
|
てっきり、大きい配列の要素を参照時にエラーが発生するかと思ったら、そんなことはなかった。
assert
とかで事前にコケるようにしておくとか必要そう。
もしくは、両者の配列のサイズが同じことを明示的に確認するのが吉。
また蛇足だが、Stackoverflow では意図的に異なる大きさの配列を上手く循環させつつ回したい場合の対処法も書いてあり勉強になった。1
2022-10-27: 追記
@ftnext さんから以下の情報2を教えてもらいました。
小さい方を読み切ったら for を抜けるの予想と違いますよね。 3.10 から zip に strict 引数が追加されており、True を指定すれば長さが異なると ValueError を送出するようになったんです! https://docs.python.org/ja/3/library/functions.html#zip… また長い方に合わせたいときは zip_longest が標準ライブラリの itertools にありますー
なるほど。
Without the
strict=True
argument, any bug that results in iterables of different lengths will be silenced, possibly manifesting as a hard-to-find bug in another part of the program.
わかる〜〜ということで試してみました。
|
|
おぉ、これは便利ですね!
結論
Python で zip 関数を使う場合、2つの配列に同じ大きさを想定する場合は strict=True を使うとバグの温床を潰せる
See Also
- Apache Beam 2.40 で導入された scikit-lean, Pytorch の効率的な推論が可能になる RunInference API を試してみる
- poetry show でパッケージ名に (!) が付与されている意味
- 社内でデータ分析結果を可視化・共有する際に Google Colab が便利
- Google Colab で Spacy による NER の結果を表示するには、jupyter 引数を True にする必要がある
- Python で DeepL API Free を利用してテキストファイルを翻訳する