Python自动完成变量名称

Question

我是Python的新手，并且希望在while循环内使用自动完成的变量。 我尝试举一个最小的例子。 假设我的文件夹中有以下文件，每个文件以相同的字母开头，并以递增的数字开头，但文件名的末尾仅由随机数组成

a_i=1_404
a_i=2_383
a_i=3_180

我想要一段时间的循环

while 1 <= 3:
    old_timestep     = 'a_i=n_*'
    actual_timestep  = 'a_i=(n+1)_*' 
    ... (some functions that need the filenames saved in the two above initialised variables)
    n = n+1

因此，如果我开始循环，则希望它自动处理目录中的所有文件。 结果，存在两个问题：

1）如何告诉python（在我的示例中，我使用了'*'）我希望文件名能够自动完成？

2）如何在文件名中使用公式（在我的示例中为“（n + 1）”）？

提前谢谢了！

Answer 1

1）据我所知，您不能自动执行此操作。 我会将所有文件名存储在列表中的目录中，然后在该列表中进行搜索

from os import listdir
from os.path import isfile, join
dir_files = [ f for f in listdir('.') if isfile(join('.',f)) ]
while i <= 3:
    old_timestep = "a_i=n_"
    for f in dir_files:
        if f.startswith(old_timestep):
            # process file
    i += 1

2）您可以使用字符串串联

f = open("a_i=" + str(n + 1) + "remainder of filename", 'w')

Answer 2

您可以使用水珠模块做*扩展：

import glob
old = next(glob.glob('a_i={}_*'.format(n)))
actual = next(glob.glob('a_i={}_*'.format(n + 1)))

Answer 3

我找到了一种方法来解决1）自己：首先做a）然后做b）：

1）a）截断文件名，以便只剩下前x个字符：

for i in {1..3}; do
    cp a_i=$((i))_* a_i=$((i))
done

b）

n = 1
while n <= 3:
    old = 'a_i=' + str(n) 
    new = 'a_i=' + str(n+1)

str（）用于将整数n转换为字符串，以连接感谢您的输入！