正在搜索文件python

Question

使用此代码 我希望它搜索名为sh的所有文件，如sh.c，sh.txt，sh.cpp等。但除非我编写此代码不会搜索lookfor = sh.txt或lookfor = sh.pdf而不是lookfor = sh在下面的代码中。 因此，我希望通过编写lookfor = sh它搜索名为sh的所有文件。请帮助。

import os 
from os.path import join 
lookfor = "sh" 
for root, dirs, files in os.walk('C:\\'): 
    if lookfor in files: 
      print "found: %s" % join(root, lookfor) 
      break 

Answer 1

替换：

if lookfor in files: 

有了：

for filename in files: 
    if filename.rsplit('.', 1)[0] == lookfor: 

---

什么filename.rsplit('.', 1)[0]是删除这是一个点（==扩展名）后，找到该文件的最右侧。如果文件中有多个点，我们将其余的文件保存在文件名中。

Answer 2

线

if lookfor in files: 

说，如果列表files包含lookfor给出的字符串下面的代码应执行。

但是，您希望测试应该是找到的文件名从给定的字符串开始并继续使用.。

此外，你想要确定真实的文件名。

所以，你的代码应该是

import os 
from os.path import join, splitext 
lookfor = "sh" 
found = None 
for root, dirs, files in os.walk('C:\\'): 
    for file in files: # test them one after the other 
     if splitext(filename)[0] == lookfor: 
      found = join(root, file) 
      print "found: %s" % found 
      break 
    if found: break 

这甚至可以改善，因为我不喜欢我怎么休息外for循环的方式。

也许你想拥有它的功能：

def look(lookfor, where): 
    import os 
    from os.path import join, splitext 
    for root, dirs, files in os.walk(where): 
     for file in files: # test them one after the other 
      if splitext(filename)[0] == lookfor: 
       found = join(root, file) 
       return found 

found = look("sh", "C:\\") 
if found is not None: 
    print "found: %s" % found 

Answer 3

import os 
from os.path import join 
lookfor = "sh." 
for root, dirs, files in os.walk('C:\\'): 
    for filename in files: 
     if filename.startswith(lookfor): 
      print "found: %s" % join(root, filename) 

您可能需要阅读的fnmatch的doc和太glob的。

Answer 4

尝试水珠：

import glob 
print glob.glob('sh.*') #this will give you all sh.* files in the current dir 

Answer 5

想必你想SH他们的基本名称向搜索文件。 （名称的部分不包括路径和扩展名）您可以使用fnmatch模块的filter功能执行此操作。

import os 
from os.path import join 
import fnmatch 
lookfor = "sh.*" 
for root, dirs, files in os.walk('C:\\'): 
    for found in fnmatch.filter(files, lookfor): 
     print "found: %s" % join(root, found)