Python进阶编程:编写更高效、优雅的Python代码
上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()方法的效果是很不错的。