上QQ阅读APP看书,第一时间看更新
2.3.1 字符串开头或结尾匹配
对于一些文件格式,我们要通过指定的文本模式去检查字符串的开头或者结尾,比如文件名后缀、URL Scheme等。
检查字符串开头或结尾的一个简单方法是使用str.startswith()或者str.endswith()方法,示例如下:
file_name = 'python.txt' print(file_name.endswith('.txt')) print(file_name.startswith('abc')) url_val = 'http://www.python.org' print(url_val.startswith('http:'))
但如果想检查多种匹配可能,需要将所有的匹配项放到一个元组中,然后传给startswith()或者endswith()方法。endswith()方法示例如下:
import os file_name_list = os.listdir('.') print(file_name_list) print([name for name in file_name_list if name.endswith(('.py', '.h')) ]) print(any(name.endswith('.py') for name in file_name_list))
startswith()方法示例如下:
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()类型,相关代码(starts_ends_with.py)示例如下:
web_pre_list = ['http:', 'ftp:'] url_val = 'http://www.python.org' print(url_val.startswith(tuple(web_pre_list))) print(url_val.startswith(web_pre_list))
执行py文件,得到类似如下的输出:
True Traceback (most recent call last): File "/Users/lyz/Desktop/python-workspace/advanced_programming/chapter2/starts_ends_with_exp.py", line 30, in <module> print(url_val.startswith(web_pre_list)) TypeError: startswith first arg must be str or a tuple of str, not list
使用startswith()和endswith()方法对字符串开头和结尾检查非常方便。类似的操作也可以使用切片来替换,但是代码看起来没有那么优雅,代码可读性也要差一些,示例如下:
file_name = 'test.txt' print(file_name[-4:] == '.txt') url_val = 'http://www.python.org' print(url_val[:5] == 'http:' or url_val[:6] == 'https:' or url_val[:4] == 'ftp:')
当然,我们也可以使用正则表达式实现。读者可以根据需要自行实现。
提示 当需要和其他操作(比如普通数据聚合)相结合的时候,配合使用startswith()和endswith()方法的效果是很不错的。