如何在Python中多次循环遍历目录中的指定文件

我来帮你搞定这个循环遍历文件的需求！首先先修正下你现有代码里的小细节：Python里用反引号i转字符串的写法已经过时啦，推荐用更清晰的f-string或者str(i)生成文件名，比如：

discharge_filename = [f"streamflow{i}.nc" for i in range(1000, 2000)]

接下来针对「遍历到最后一个文件后自动回到第一个重新开始」的需求，有两种简单好用的实现方式：

方法1：用itertools.cycle实现无限循环迭代 #

Python标准库的itertools.cycle能帮你创建一个无限循环的迭代器，它会反复遍历你的文件名列表，完全不用自己处理循环逻辑，非常省心：

import itertools

# 生成文件名列表
discharge_filename = [f"streamflow{i}.nc" for i in range(1000, 2000)]
# 创建循环迭代器
file_iterator = itertools.cycle(discharge_filename)

# 假设你的模型需要运行nt个时间步
nt = 5000  # 替换成你实际的时间步数
for _ in range(nt):
    current_file = next(file_iterator)
    # 在这里把current_file的数据输入模型，比如：
    # load_data(current_file)
    # model.run_step()
    print(f"当前时间步处理文件: {current_file}")

这种方法的优势是代码简洁，适合不需要手动控制索引的场景，迭代器会自动帮你循环往复。

方法2：手动用取模运算控制索引 #

如果你更倾向于自己掌控循环逻辑，可以用取模运算%实现循环索引，每次用当前时间步对文件总数取模，就能得到循环后的文件索引：

# 生成文件名列表
discharge_filename = [f"streamflow{i}.nc" for i in range(1000, 2000)]
num_files = len(discharge_filename)
nt = 5000  # 实际时间步数

for step in range(nt):
    # 计算当前循环的文件索引
    file_idx = step % num_files
    current_file = discharge_filename[file_idx]
    # 处理文件并输入模型
    print(f"时间步{step}处理文件: {current_file}")

这种方法的好处是你可以随时获取当前的循环次数或者索引，适合需要额外逻辑控制的场景。

两种方法都能完美实现你要的「多次循环遍历文件」的需求，你可以根据自己的代码习惯选择～

内容的提问来源于stack exchange，提问作者B_Lynch