Python startswith()、endswith()

目录

• 一、问题
• 二、解决方案

检查字符串的开头或结尾。

二、解决方案

startswith()、endswith()。

filename = 'test.txt'
print(filename.endswith('.txt'))
print(filename.startswith('file.'))

输出：

True
False

检查多种匹配，将这些匹配放到一个元组中。

import os

filenames = os.listdir('.')     # 当前目录下的文件和文件夹
print(filenames)

输出：

['decision_tree_model.py', 
 'gbdt_classifier_example.py', 
 'gbdt_model.py', 
 'utils', 
 'test.txt'
 '__pycache__']

name_lst = [name for name in filenames if name.endswith(('.py', '.txt'))]
print(name_lst)
print()

输出：

['decision_tree_model.py', 
 'gbdt_classifier_example.py', 
 'gbdt_model.py', 
 'test.txt']

startswith()、endswith()常与普通聚合函数结合使用。

print(any(name.endswith('.py') for name in filenames))

输出：

True