Python单函数读取常用文件

代码如下：

import pandas as pd


# 读取数据文件
def readDataFile(readPath):  # readPath: 数据文件的地址和文件名
    try:
        if (readPath[-4:] == ".csv"):
            dfFile = pd.read_csv(readPath, header=0, sep=",")  # 间隔符为逗号，首行为标题行
            # dfFile = pd.read_csv(filePath, header=None, sep=",")  # sep: 间隔符，无标题行
        elif (readPath[-4:] == ".xls") or (readPath[-5:] == ".xlsx"):  # sheet_name 默认为 0
            dfFile = pd.read_excel(readPath, header=0)  # 首行为标题行
            # dfFile = pd.read_excel(filePath, header=None)  # 无标题行
        elif (readPath[-4:] == ".dat"):  # sep: 间隔符，header：首行是否为标题行
            dfFile = pd.read_table(readPath, sep=" ", header=0)  # 间隔符为空格，首行为标题行
            # dfFile = pd.read_table(filePath,sep=",",header=None) # 间隔符为逗号，无标题行
        else:
            print("不支持的文件格式。")
    except Exception as e:
        print("读取数据文件失败：{}".format(str(e)))
        return
    return dfFile


if __name__ == '__main__':
    readPath = "C:/Users/lenovo/Desktop/lut.xls"  # 数据文件的地址和文件名
    dfFile = readDataFile(readPath)  # 调用读取文件子程序

    print(type(dfFile))  # 查看 dfFile 数据类型
    print(dfFile.shape)  # 查看 dfFile 形状（行数，列数）
    print(dfFile.head())  # 显示 dfFile 前 5 行数据

代码讲解

1、readPath[-4:]是对字符串切片，选取了倒数4位，也就是.以及.后面的文件格式。

2、read_csv是pd下读取.csv格式文件的函数。主要包括三个属性（路径，header，sep），路径是读取文件存放的路径，

3、header=0的含义，下图为源文件，
在这里插入图片描述
当设置header=0时表示首行为标题行，我们打印读取结果的前五行数据会发现自动读取了标题

当设置header=1时，将第一行数据行作为读取数据的标题行，因此整体数据会少了一行，从500变为499

当header不设置时，默认第0行为标题行
在这里插入图片描述
当设置为header=None时，会认为原有数据没有标题行，并自动生成一个0开始的标题行。