字符串開(kāi)頭或結(jié)尾匹配

問(wèn)題

你需要通過(guò)指定的文本模式去檢查字符串的開(kāi)頭或者結(jié)尾，比如文件名后綴， URL
Scheme 等等。

解法

檢查字符串開(kāi)頭或結(jié)尾的一個(gè)簡(jiǎn)單方法是使用 str.startswith() 或者是str.endswith() 方法。比如：

>>> filename = 'spam.txt' >>> filename.endswith('.txt') True >>> filename.startswith('file:') False >>> url = 'http://www.python.org' >>> url.startswith('http:') True >>>

如果你想檢查多種匹配可能，只需要將所有的匹配項(xiàng)放入到一個(gè)元組中去，然后傳給 startswith() 或者 endswith() 方法：

>>> import os >>> filenames = os.listdir('.') >>> filenames [ 'Makefile', 'foo.c', 'bar.py', 'spam.c', 'spam.h' ] >>> [name for name in filenames if name.endswith(('.c', '.h')) ] ['foo.c', 'spam.c', 'spam.h' >>> any(name.endswith('.py') for name in filenames) True >>>

另一個(gè)例子

from urllib.request import urlopen def read_data(name):if name.startswith(('http:', 'https:', 'ftp:')):return urlopen(name).read()else:with open(name) as f:return f.read()

奇怪的是，這個(gè)方法中必須要輸入一個(gè)元組作為參數(shù)。如果你恰巧有一個(gè) list 或者 set 類型的選擇項(xiàng)，要確保傳遞參數(shù)前先調(diào)用 tuple() 將其轉(zhuǎn)換為元組類型。比如：

>>> choices = ['http:', 'ftp:'] >>> url = 'http://www.python.org' >>> url.startswith(choices) Traceback (most recent call last): File "<stdin>", line 1, in <module> TypeError: startswith first arg must be str or a tuple of str, not list >>> url.startswith(tuple(choices)) True >>>

討論

startswith() 和 endswith() 方法提供了一個(gè)非常方便的方式去做字符串開(kāi)頭和結(jié)尾的檢查。類似的操作也可以使用切片來(lái)實(shí)現(xiàn)，但是代碼看起來(lái)沒(méi)有那么優(yōu)雅。比如：

>>> filename = 'spam.txt' >>> filename[-4:] == '.txt' True >>> url = 'http://www.python.org' >>> url[:5] == 'http:' or url[:6] == 'https:' or url[:4] == 'ftp:' True >>>

你可以能還想使用正則表達(dá)式去實(shí)現(xiàn)，比如：

>>> import re >>> url = 'http://www.python.org' >>> re.match('http:jhttps:jftp:', url) <_sre.SRE_Match object at 0x101253098> >>>

這種方式也行，但是對(duì)于簡(jiǎn)單的匹配實(shí)在是有點(diǎn)殺雞用牛刀了，本節(jié)中的方法更加簡(jiǎn)潔。

最后提一下，當(dāng)和其他操作比如普通數(shù)據(jù)聚合相結(jié)合的時(shí)候 startswith() 和 endswith() 方法是很不錯(cuò)的。比如，下面這個(gè)語(yǔ)句檢查某個(gè)文件夾中是否存在指定的文件類型：

if any(name.endswith(('.c', '.h')) for name in listdir(dirname)): ...

總結(jié)

以上是生活随笔為你收集整理的《Python Cookbook 3rd》笔记（2.2）：字符串开头或结尾匹配的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。