common层-解析xls、csv文件的方法

最新推荐文章于 2024-05-11 20:00:00 发布

Amir_zy

最新推荐文章于 2024-05-11 20:00:00 发布

阅读量936

点赞数 6

文章标签： selenium 测试工具

本文链接：https://blog.csdn.net/zy229544/article/details/135597134

版权

'''
作用: 解析xls, csv文件
'''
import pandas

class ReadFile:
    def read_csv_excel(self,path):
        if path.endswith('xls') or path.endswith('xlsx'):
            file = pandas.read_excel(path)
        elif path.endswith('csv'):
            file = pandas.read_csv(path)
        else:
            return '文件格式无法处理'

        data = []
        for i in file.index.values:  # 遍历所有的索引号
            # 通过索引号去读取, 并转换成字典
            data_dict = file.loc[i].to_dict()
            # 每次循环都把这个字典加到上面的列表里
            data.append(data_dict)
        return data

    def read_txt(self, path):
        data_dict = {}
        with open(path, 'r', encoding='utf8') as f:
            lines = f.read().strip().split('\n')
            for line in lines:
                content = line.strip().split('=')
                data_dict[content[0]] = content[1]
        return data_dict

if __name__ == '__main__':
    readfile = ReadFile()
    data = readfile.read_csv_excel(r'E:\A\python\project\chap1\dayzy1\pandas_data\data_csv.csv')
    print(data)