startswith() 和 endswith() 方法提供了一个非常方便的方式去做字符串开头和结尾的检查。
1、查看指定目录下的所有文件名
>>> import os
>>> filenames = os.listdir('I:\PythonTest')
>>> filenames
['111.csv', '111.xlsx', '111.xml', '123.txt', '123.xlsx', '123123.xml', '123123.xml.bak', '1234.txt', '222.xml', 'book.xml', 'book.xml.bak', 'excelWrite.csv', 'excelWrite.xlsx', 'Koala.jpg', 'movie.xml', 'movie.xml.bak', 'movies.xml', 'receive.txt', 'user.xml', 'user.xml.bak', '新建文件夹']
2、列出.txt文件名
>>> for i in filenames:
if i.endswith('.txt'):
print(i)
123.txt
1234.txt
receive.txt
另外一种写法:
>>> i for i in filenames if i.endswith('.txt')