《Python Cookbook 3rd》笔记（2.2）：字符串开头或结尾匹配

古城微笑少年丶 2022-12-20 02:29 199阅读 0赞

字符串开头或结尾匹配

问题

你需要通过指定的文本模式去检查字符串的开头或者结尾，比如文件名后缀， URL
Scheme 等等。

解法

检查字符串开头或结尾的一个简单方法是使用 str.startswith() 或者是str.endswith() 方法。比如：

  1. >>> filename = 'spam.txt'
  2. >>> filename.endswith('.txt')
  3. True
  4. >>> filename.startswith('file:')
  5. False
  6. >>> url = 'http://www.python.org'
  7. >>> url.startswith('http:')
  8. True
  9. >>>

如果你想检查多种匹配可能，只需要将所有的匹配项放入到一个元组中去，然后传给 startswith() 或者 endswith() 方法：

  1. >>> import os
  2. >>> filenames = os.listdir('.')
  3. >>> filenames
  4. [ 'Makefile', 'foo.c', 'bar.py', 'spam.c', 'spam.h' ]
  5. >>> [name for name in filenames if name.endswith(('.c', '.h')) ]
  6. ['foo.c', 'spam.c', 'spam.h'
  7. >>> any(name.endswith('.py') for name in filenames)
  8. True
  9. >>>

另一个例子

  1. from urllib.request import urlopen
  2. def read_data(name):
  3.     if name.startswith(('http:', 'https:', 'ftp:')):
  4.         return urlopen(name).read()
  5.     else:
  6.         with open(name) as f:
  7.             return f.read()

奇怪的是，这个方法中必须要输入一个元组作为参数。如果你恰巧有一个 list 或者 set 类型的选择项，要确保传递参数前先调用 tuple() 将其转换为元组类型。比如：

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

讨论

startswith() 和 endswith() 方法提供了一个非常方便的方式去做字符串开头和结尾的检查。类似的操作也可以使用切片来实现，但是代码看起来没有那么优雅。比如：

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

你可以能还想使用正则表达式去实现，比如：

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

这种方式也行，但是对于简单的匹配实在是有点杀鸡用牛刀了，本节中的方法更加简洁。

最后提一下，当和其他操作比如普通数据聚合相结合的时候 startswith() 和 endswith() 方法是很不错的。比如，下面这个语句检查某个文件夹中是否存在指定的文件类型：

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

发表评论

表情：
评论列表 （有 0 条评论，199人围观）

还没有评论，来说两句吧...

相关阅读