根据输入列表 Python 重命名文件

Question

我在一个目录中有一堆文件。它们会使用以下名称自动下载：

new.pdb
new(1).pdb
new(2).pdb and so on

我有一个实际文件名列表，例如：

AAA
BBB
CCC

我想要一种方法来重命名列表中相应名称的文件，其中 new.pdb 将是 AAA.pdb，new(1).pdb 将是 BBB.pdb 等等。 如果您能在投反对票之前给我一个提示，我将不胜感激。我已经尽力解决这个问题，但不能。 到目前为止，我已经尝试过了。但是以下代码的问题是，一些文件的下载速度可能比在它们之前开始的文件要快。所以它不会像我想要的那样重命名。

    infiles = filter(os.path.isfile, os.listdir(os.getcwd()))
    infiles = [os.path.join(os.getcwd(), f) for f in infiles] 
    infiles.sort(key=lambda x: os.path.getmtime(x))
    last_file = infiles[-1]
    if rename_files.endswith(".pdb"):
        os.rename(last_file, directory_name+".pdb")

Answer 1

前三行是正确的。

但是，要使for <variable> in <collection> 构造起作用，<collection> 必须是一个集合。 last_file 是字符串，因此是字符的集合；并且单个字符不可能以.pdb 结尾。

假设rename_files 是新名称列表，您希望在前三行之后使用它：

infiles = (infile for infile in infiles if infile.endswith(".pdb"))
for infile, outfile in zip(infiles, rename_files):
    os.rename(infile, outfile + ".pdb")

zip 将从列表中创建一个配对列表，因此您首先找到所有 pdb 文件，将它们与新名称配对，然后根据配对重命名每个文件。

编辑：如果您想使用文件名而不是原始 mtime，则需要提取该数字并按其排序：

def get_rank(infile):
    m = re.search(r'(?:\((\d+)\))[^()]*', infile)
    if m:
        return int(m.group(1))
    return 0 
infiles = filter(os.path.isfile, os.listdir(os.getcwd()))
infiles = [os.path.join(os.getcwd(), f) for f in infiles if f.endswith('.txt')]
filerank = (get_rank(infile) for infile in infiles)
renames = zip(filerank, infiles, rename_files)
renames.sort(key=lambda x: x[0])
for _, infile, outfile in renames:
    os.rename(infile, outfile + ".pdb")