python 获取指定类型文件
#!/usr/bin/env python
# encoding: utf-8
# coding style: pep8
# ====================================================
# Copyright (C)2019 All rights reserved.
#
# Author :
# Email :
# File Name : listdir.py
# Last Modified : 2019-12-31 15:25
# Description : 获取当前目录下的txt文本文件
#
# ====================================================
import sys
import os
dirs = os.listdir("./")
print("当前目录中的所有文件:\n",dirs)
print("包含的txt文件:")
for file_name in dirs:
if os.path.splitext(file_name)[1] == '.txt':
print(file_name)
- os.listdir(“dir_name”) 返回一个列表,包含当前目录(不递归)中的所有文件及文件夹,包含所有隐藏文件及文件夹,但是有两个特殊文件’.‘和’…'除外。
- os.path.splitext(“file name”) 分离文件名及后缀,即:(file_prefix,file_suffix)
比如当前目录
.
├── a
│ └── a.txt
├── b
├── filename
├── listdir.py
├── vimtest1.txt
├── vimtest2.txt
└── vimtest3.txt
2 directories, 6 files
运行结果:
当前目录中的所有文件:
[‘vimtest2.txt’, ‘.DS_Store’, ‘vimtest3.txt’, ‘vimtest1.txt’, ‘a’, ‘filename’, ‘listdir.py’, ‘b’]
包含的txt文件:
vimtest2.txt
vimtest3.txt
vimtest1.txt