python - 选择以给定字符串开头的文件

Question

在一个目录中，我有很多文件，名称或多或少是这样的：

001_MN_DX_1_M_32
001_MN_SX_1_M_33
012_BC_2_F_23
...
...

在 Python 中，我必须编写一个代码，从目录中选择一个以某个字符串开头的文件。例如，如果字符串是001_MN_DX，Python 选择第一个文件，依此类推。

我该怎么做？

Answer 1

import os
prefixed = [filename for filename in os.listdir('.') if filename.startswith("prefix")]

Answer 2

尝试使用os.listdir和。 长格式（带有 for 循环），os.path.joinos.path.isfile

import os
path = 'C:/'
files = []
for i in os.listdir(path):
    if os.path.isfile(os.path.join(path,i)) and '001_MN_DX' in i:
        files.append(i)

带有列表理解的代码是

import os
path = 'C:/'
files = [i for i in os.listdir(path) if os.path.isfile(os.path.join(path,i)) and \
         '001_MN_DX' in i]

在这里查看详细的解释......

Answer 3

您可以使用模块glob，它遵循 Unix shell 规则进行模式匹配。 看更多。

from glob import glob

files = glob('*001_MN_DX*')

Answer 4

import os, re
for f in os.listdir('.'):
   if re.match('001_MN_DX', f):
       print f

Answer 5

您可以使用 os 模块列出目录中的文件。

例如：查找当前目录下所有名称以 001_MN_DX 开头的文件

import os
list_of_files = os.listdir(os.getcwd()) #list of files in the current directory
for each_file in list_of_files:
    if each_file.startswith('001_MN_DX'):  #since its all type str you can simply use startswith
        print each_file

Answer 6

使用较新的pathlib模块，请参阅链接：

from pathlib import Path
myDir = Path('my_directory/')

fileNames = [file.name for file in myDir.iterdir() if file.name.startswith('prefix')]    
filePaths = [file for file in myDir.iterdir() if file.name.startswith('prefix')]

Answer 7

import os
 for filename in os.listdir('.'):
    if filename.startswith('criteria here'):
        print filename #print the name of the file to make sure it is what 
                               you really want. If it's not, review your criteria
                #Do stuff with that file