Pythonでループする：最初の繰り返しの前に行うこと

Question

私は最適化したい。

シンプルなソリューション

connection = get_db_connection() 
for item in my_iterator: 
    push_item_to_db(item, connection) 

欠点：

get_db_connection()が遅いです。 my_iteratorが空の場合、私はそれを呼ぶことを避けたいと思います。

ソリューション

connection = None 
for item in my_iterator: 
    if connection is None: 
     connection = get_db_connection() 
    push_item_to_db(item, connection) 

欠点 "なしの場合"：

100kの項目がmy_iteratorである場合（それは一度だけ必要とされているが）、その後、if connection is Noneが100K回呼び出されます。私はこれを避けたい。

完璧なソリューション...

1. は、イテレータが
2. 空の場合は、すべての反復のために無駄if connection is None:を呼び出すことはありませんget_db_connection()を呼び出すことはありません。

connection = None 
for item in my_iterator: 
    if connection is None: 
     connection = get_db_connection() 
    push_item_to_db(item, connection) 

シンプルなソリューション：

Answer 1

あなたのような何かを行うことができます。それを考え直す必要はありません。 100Kの操作でも、x is Noneは、Pythonのオペコードを1つ取るだけの参照比較です。すべての挿入時に発生する完全なtcpラウンドトリップ+ディスク書き込みと比較して、これを最適化する必要はありません。それはおそらく本当である

def put_items_to_database (iterator): 
    try: 
     item = next(iterator) 

     # We connect to the database only after we 
     # know there at least one element in the collection    
     connection = get_db_connection() 

     while True: 
      push_item_to_db(item, connection) 
      item = next(iterator) 
    except StopIteration: 
     pass 

：

Answer 2

あなたは私は、Pythonの専門家ではないですコード

if (len(my_iterator)>0): 
    connection = get_db_connection() 
    for item in my_iterator: 
     push_item_to_db(item, connection) 

Answer 3

の全体のセクションの前にイテレータ・カウントをチェックすることができますが、私はこのような何かをするだろうパフォーマンスはここでデータベースに結び付けられています。しかし、疑問は、不必要な作業を避ける方法を見つけることであり、上記は反復中に何が起こるかを正確に制御する基本的な方法です。

他の解決策は何らかの形で「よりシンプル」ですが、一方で私はこれがより明白であり、最小の驚きの原則に従うと考えています。

Answer 4

ソリューションは、1

これはwhile Trueループなしで動作します。

try: 
    next(my_iterator) 
    connection = get_db_connection() 
    push_item_to_db(item, connection) 
except StopIteration: 
    pass 
for item in my_iterator: 
    push_item_to_db(item, connection) 

ソリューション2

あなたはそのイテレータがNone（またはその他のユニークなオブジェクト）を返すことはありませんことがわかっている場合、あなたはnext()のデフォルトを利用することができます：

if next(my_iterator, None) is not None: 
    connection = get_db_connection() 
    push_item_to_db(item, connection) 
for item in my_iterator: 
    push_item_to_db(item, connection) 

ソリューション3

イテレータによって返されない値を保証できない場合は、センチネルを使用できます。

sentinel = object() 
if next(my_iterator, sentinel) is not sentinel: 
    connection = get_db_connection() 
    push_item_to_db(item, connection) 
for item in my_iterator: 
    push_item_to_db(item, connection) 

ソリューションitertools.chain()を使用して4

：

from itertools import chain 

for first_item in my_iterator: 
    connection = get_db_connection() 
    for item in chain([first_item], my_iterator): 
     push_item_to_db(item, connection) 

Answer 5

for item in my_iterator: 
    # First item (if any) 
    connection = get_db_connection() 
    push_item_to_db(item, connection) 
    for item in my_iterator: 
     # Next items 
     push_item_to_db(item, connection)