字符串開頭或結尾匹配

問題

你需要通過指定的文本模式去檢查字符串的開頭或者結尾，比如文件名后綴， URL
Scheme 等等。

解法

檢查字符串開頭或結尾的一個簡單方法是使用 str.startswith() 或者是str.endswith() 方法。比如：

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

如果你想檢查多種匹配可能，只需要將所有的匹配項放入到一個元組中去，然后傳給 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 >>>

另一個例子

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()

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

>>> 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() 方法提供了一個非常方便的方式去做字符串開頭和結尾的檢查。類似的操作也可以使用切片來實現，但是代碼看起來沒有那么優雅。比如：

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

你可以能還想使用正則表達式去實現，比如：

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

這種方式也行，但是對于簡單的匹配實在是有點殺雞用牛刀了，本節中的方法更加簡潔。

最后提一下，當和其他操作比如普通數據聚合相結合的時候 startswith() 和 endswith() 方法是很不錯的。比如，下面這個語句檢查某個文件夾中是否存在指定的文件類型：

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。