正在搜索文件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)