Python 過濾字符串的技巧,map與itertools.imap

2019-11-25 18:46:17

字體：大中小

供稿：網(wǎng)友

具體的實例
我們需要在目錄中遍歷，包括子目錄（哈哈），找出所有后綴為：rmvb ，avi ，pmp 的文件。（天哪？！你要干什么？這可是我的隱私啊～～）

import os 

def anyTrue(predicate, sequence): 
return True in map(predicate, sequence) 

def filterFiles(folder, exts): 
for fileName in os.listdir(folder): 
if os.path.isdir(folder + '/' + fileName): 
filterFiles(folder + '/' + fileName, exts) 
elif anyTrue(fileName.endswith, exts): 
print fileName 

exts = ['.rmvb', '.avi', '.pmp'] 
filterFiles('/media/Personal/Movie', exts) 

輸出結(jié)果
來看看有什么好東東：
[66影視www.66ys.cn]迷失第四季04.rmvb
[迷失.第4季].Lost.S04E00.rmvb
[迷失Lost第四季][第02集][中文字幕].rmvb
《迷失Lost第四季》第05集[中文字幕].rmvb
《迷失Lost第四季》第06集[中文字幕].rmvb
《迷失Lost第四季》第07集[中文字幕].rmvb
天賜第2季01.rmvb
天賜第2季02.rmvb
天賜第2季03.rmvb
天賜第2季04.rmvb
天賜第2季05.rmvb
影視帝國(bbs.cnxp.com).美麗心靈.A.Beautiful.Mind.2001.CD1.rmvb
( ... 太多了，不要全輸出來吧～～)

擴展
CookBook一書中，提供的是itertools.imap來實現(xiàn)對字符串的過濾。imap和map不同的是，imap返回的是一個iteration對象，而map返回的是一個list對象。代碼如下：
import itertools
def anyTrue(predicate, sequence):
return True in itertools.imap(predicate, sequence)
def endsWith(s, *endings):
return anyTrue(s.endswith, endings)
imap 等價于：
def imap(function, *iterables):
iterables = map(iter, iterables)
while True:
args = [i.next() for i in iterables]
if function is None:
yield tuple(args)
else:
yield function(*args)

上一篇：Python open讀寫文件實現(xiàn)腳本

下一篇：比較詳細Python正則表達式操作指南(re使用)