python每日一练之读取文件到字典

学python最重要的是练习,学了不练假把式。每天找点有意思的练习题做做,强化python技能很有裨益。
需求:现有list.txt文件存放了商品销售数据,需要读取该文件到字典列表。

​ 文件内容:
​ 商品,单价,数量
​ 苹果,4.8,5
​ 雪梨,2.9,10
​ 大枣,5.8,6

​ 预期结果是:
​ [{'商品': '苹果', '单价': '4.8', '数量': '5'},
​ {'商品': '雪梨', '单价': '2.9', '数量': '10'},
​ {'商品': '大枣', '单价': '5.8', '数量': '6'}]

先做分析:文件是文本文件,uf8编码,用open默认r模式打开即可。第一行内容判断为字典的key,后面的行是value。

with open('a.txt', encoding='utf8') as f:
    name = f.readline().strip().split(',')  # 读取文件第一行转成list,作为字典的key
    fruit_list = []
    for line in f:  # 用for循环遍历句柄f,优点是无论文件有多大,读取都不会撑爆内存。
        # 不要用read()或readlines(),万一处理的文件超大会导致撑爆内存。
        if len(line) < 3:  # 过滤空行及数据不完整的行。
            continue
        line = line.strip().split(',')  # 将文件内容按','分隔转成列表
        fruit_dict = {}  # 声明一空字典,保存每一行的内容
        for i in range(len(name)):  # 通过下标遍历name列表
            fruit_dict[name[i]] = line[i]
        fruit_list.append(fruit_dict)
    print(fruit_list)

写完感觉不完美,最好是封装成函数,利于代码复用。

最终版本是:

def file_to_list(file):
    '''
    @param file: 参数是文件名,文件格式是以','分隔的txt文件。
    @return: 返回一个列表,列表元素是字典,内容源自file文件。
    '''

    with open(file, encoding='utf8') as f:
        name = f.readline().strip().split(',')  # 读取文件第一行作为字典的key
        fruit_list = []
        for line in f: 
            if len(line) < 3:  # 过滤空行及数据不完整的行。
                continue
            line = line.split(',') 
            fruit_dict = {} 
            for i in range(len(name)):  # 遍历name列表,将每行对应内容做成字典
                fruit_dict[name[i]] = line[i].strip()
            fruit_list.append(fruit_dict) 
    return fruit_list

至此,读取文件到字典练习完毕。

总结:
一是在开始学python时就要养成好习惯,将功能模块化,提高代码复用率和可读性。
二是定义方法要写文档注释,方法参数和返回值要解释清楚。
三是代码注释不用每行都写,重点是讲清楚代码逻辑。

你可能感兴趣的:(Python,列表,python,java,csv,linux)