python のイテレータを再利用可能なリストへ変換

python のファイル一覧取得 os.scandir をループで使用して痛い目に遭う

view plainprint?

1. def test(dir):  
2.   files = os.scandir(dir)  
3.   
4.   print('list1')  
5.   for file in files:  
6.       print(file.path)  
7.   
8.   print('list2')  
9.   for file in files:  
10.       print(file.path)  

  def test(dir):
    files = os.scandir(dir)

    print('list1')
    for file in files:
        print(file.path)

    print('list2')
    for file in files:
        print(file.path)

上記コードは、二度目のループでは中身が空で何も表示されない

os.scandir で生成したリスト(イテレータ)を使いまわししてたら動きがおかしいので調べたら、戻り値はイテレータなので、片方向リンクリストのように再利用時はインデックスが進んでしまっていて戻せないんだとか

os.scandir が FindFirstFile API を利用していたら、その方が合理的ではある

書き直すの面倒だなぁと思ってたら list関数でリストへ置き換えできるそうで、小変更で済みました

view plainprint?

1. def test(dir):  
2.     files = list(os.scandir(dir))  
3.   
4.     print('list1')  
5.     for file in files:  
6.         print(file.path)  
7.   
8.     print('list2')  
9.     for file in files:  
10.         print(file.path)  

def test(dir):
    files = list(os.scandir(dir))

    print('list1')
    for file in files:
        print(file.path)

    print('list2')
    for file in files:
        print(file.path)