如何在Python中匹配字符串开头或结尾的文本？
问题..
假设您需要检查特定文本模式的字符串开头或结尾。通用模式可能是文件扩展名，但也可以是任何东西。我将向您展示几种方法。
Startswith()方法
一种检查字符串开头的简单方法是使用startswith()方法。
示例
text = "Is USA colder than Australia?"
print(f"output \n {text.startswith('Is')}")输出结果
True
示例
filename = "Hello_world.txt"
print(f"output \n {filename.startswith('Hello')}")输出结果
True
示例
site_url = 'https://www.something.com'
print(f"output \n {site_url.startswith('http:')}")输出结果
False
示例
print(f"output \n {site_url.startswith('https:')}")输出结果
True
endswith()方法。
一种检查字符串结尾的简单方法是使用endswith()方法。
输出结果
text = "Is USA colder than Australia?"
print(f"output \n {text.endswith('?')}")输出结果
True
示例
filename = "Hello_world.txt"
print(f"output \n {filename.endswith('.txt')}")输出结果
True
现在，如果要使用上述方法检查多个选择，则需要提供元组。一种常见用法是检查文件扩展名，让我们说我们需要验证目录中的“.txt”和“.csv”文件。
import os
filenames = os.listdir('.')# Let us first check if there are files
print(f"output \n {any(name.endswith(('.csv',',txt')) for name in filenames)}")输出结果
True
输出结果
[name for name in filenames if name.endswith(('.csv', '.txt')) ]输出结果
['file1.csv', 'HRDataset.csv', 'Input.csv', 'input.txt', 'input_copy.txt', 'movies_data.csv', 'my_html_data_to_csv.csv', 'temporary_file1_for_zip.csv', 'temporary_file2_for_zip.csv', 'test.csv', 'test1.txt', 'test2.txt', 'tmdb_5000_movies.csv']
请记住，这些方法接受元组，如果您要搜索的选项列表，那么我们需要将它们转换为元组。
import os
# list with choices
patters = ['.csv','.txt']
# get the file names
filenames = os.listdir('.')
# Let us first check if there are files
any(name.endswith(patters) for name in filenames)输出结果
--------------------------------------------------------------------------- TypeError Traceback (most recent call last) in 8 9 # Let us first check if there are files ---> 10 any(name.endswith(patters) for name in filenames) in (.0) 8 9 # Let us first check if there are files ---> 10 any(name.endswith(patters) for name in filenames) TypeError: endswith first arg must be str or a tuple of str, not list
上面的命令返回了一个错误，因此我们需要将列表转换成元组。
示例
# Let us first check if there are files any(name.endswith(tuple(patters)) for name in filenames)
输出结果
True
同样，我们需要将list转换为元组以获取文件名。
示例
[name for name in filenames if name.endswith(tuple(patters)) ]
输出结果
['file1.csv', 'HRDataset.csv', 'Input.csv', 'input.txt', 'input_copy.txt', 'movies_data.csv', 'my_html_data_to_csv.csv', 'temporary_file1_for_zip.csv', 'temporary_file2_for_zip.csv', 'test.csv', 'test1.txt', 'test2.txt', 'tmdb_5000_movies.csv']
最后，startswith()和endswith()方法与其他操作（例如通用数据缩减）结合使用时看起来不错。例如：
示例
if any(name.endswith(tuple(patters)) for name in filenames): <perform the logic here>