使用python读取指定目录下的指定类型文件

准备工作：设置指定的路径，使用os.listdir() 方法获取路径下所有的文件

import os
path = "d:\\data"                           # 设置路径
dirs = os.listdir(path)                    # 获取指定路径下的文件

循环判断：使用os.path.splitext()方法筛选出指定类型的文件

for i in dirs:                             # 循环读取路径下的文件并筛选输出
    if os.path.splitext(i)[1] == ".csv":   # 筛选csv文件
        print i                            # 输出所有的csv文件

案例展示：

# encoding: utf-8
import os

path = "d:\\data"                           # 设置路径
dirs = os.listdir(path)                    # 获取指定路径下的文件
for i in dirs:                             # 循环读取路径下的文件并筛选输出
    if os.path.splitext(i)[1] == ".csv":   # 筛选csv文件
        print i                            # 输出所有的csv文件

运行结果：

20160904.csv
20160911.csv
20160918.csv
20160925.csv
20161002.csv
20161009.csv

---

函数解释：
os.listdir(path)

函数功能：返回一个列表，其中包含由path指定的目录中的条目的名称。 列表是任意顺序的。它不包括特殊条目’.‘ 和’..‘，即使它们存在于目录中。

import os, sys

path = "d:\\tmp\\"
dirs = os.listdir( path )

for file in dirs:
   print (file)

运行结果：

Applicationdocs.docx
test.java
book.zip
foo.txt
Java Multiple Inheritance.html
Java Multiple Inheritance_files
java.ppt
ParallelPortViewer

os.path.splitext(path)

函数功能：分离文件名与扩展名；默认返回(fname,fextension)元组，可做切片操作

import os, sys
path = 'c:\\csv\\test.csv'
print os.path.splitext(path) 

运行结果：

('c:\\csv\\test', '.csv')