python - 将文件名与文件夹名匹配，然后移动文件

Question

我有名为“a1.txt”、“a2.txt”、“a3.txt”、“a4.txt”、“a5.txt”等的文件。然后我有名为“a1_1998”、“a2_1999”、“a3_2000”、“a4_2001”、“a5_2002”等的文件夹。

例如，我想在文件“a1.txt”和文件夹“a1_1998”之间建立连接。（我猜我需要一个正则表达式来做到这一点）。然后使用shutil将文件“a1.txt”移动到文件夹“a1_1998”，文件“a2.txt”移动到文件夹“a2_1999”等......

我是这样开始的，但由于对常规表达式缺乏了解，我陷入了困境。

import re
##list files and folders

r = re.compile('^a(?P') 
m = r.match('a') 
m.group('id')

##
##Move files to folders

我稍微修改了下面的答案以使用shutil移动文件，成功了！

import shutil
import os
import glob 

files = glob.glob(r'C:\Wam\*.txt')

for file in files: 
    # this will remove the .txt extension and keep the "aN"  
    first_part = file[7:-4]
    # find the matching directory 
    dir = glob.glob(r'C:\Wam\%s_*/' % first_part)[0]
    shutil.move(file, dir)

Answer 1

您不需要正则表达式。

像这样的东西怎么样：

import glob
files = glob.glob('*.txt')
for file in files:
    # this will remove the .txt extension and keep the "aN" 
    first_part = file[:-4]
    # find the matching directory
    dir = glob.glob('%s_*/' % first_part)[0]
    os.rename(file, os.path.join(dir, file))

Answer 2

考虑到 Inbar Rose 的建议，一个轻微的替代方案。

import os
import glob

files = glob.glob('*.txt')
dirs = glob.glob('*_*')

for file in files:
  filename = os.path.splitext(file)[0]
  matchdir = next(x for x in dirs if filename == x.rsplit('_')[0])
  os.rename(file, os.path.join(matchdir, file))