本文旨在提供一种更清晰、更易于维护的方式,来修改文件中特定行的数值。通过将文件读取、数据处理和文件写入分离成独立的函数,可以有效提高代码的可读性和可重用性。我们将展示如何使用 Python 将文本文件转换为易于操作的数据结构,修改数据,并将修改后的数据写回文件。
代码结构优化
原始代码存在一些问题,例如在读取文件的同时进行写入,这可能导致数据损坏。此外,代码逻辑较为复杂,可读性较差。为了解决这些问题,我们将代码分解为三个独立的函数:getBooks、decrementBookUnits 和 writeBooks。
1. getBooks(file):读取文件并转换为数据结构
此函数负责读取文件内容,并将其转换为一个易于操作的 Python 列表,其中每个元素都是一个字典,代表一本书的信息。
def getBooks(file): """ 读取文件内容,并将其转换为一个列表,其中每个元素都是一个字典,代表一本书的信息。 :param file: 文件名或文件路径。 :return: 一个列表,其中每个元素都是一个字典,代表一本书的信息。 """ with open(file, 'r') as booksFile: content = booksFile.read().splitlines() lines=[line.split(' -- ') for line in content] books=[dict(entry.split(':') for entry in line) for line in lines] return books
代码解释:
立即学习“Python免费学习笔记(深入)”;
- with open(file, ‘r’) as booksFile:: 使用 with 语句打开文件,确保文件在使用完毕后自动关闭。
- content = booksFile.read().splitlines(): 读取文件所有内容,并按行分割成一个字符串列表。
- lines=[line.split(‘ — ‘) for line in content]: 将每一行按照 ” — ” 分割成键值对列表。
- books=[dict(entry.split(‘:’) for entry in line) for line in lines]: 将每个键值对列表转换为字典,最终生成一个包含所有书籍信息的字典列表。
2. decrementBookUnits(books, isbn):修改书籍信息
此函数接收书籍列表和 ISBN 作为输入,并在列表中找到匹配的书籍,然后修改其 ISBN 的值。
def decrementBookUnits(books, isbn): """ 修改书籍列表中指定 ISBN 的书籍的 ISBN 值。 :param books: 书籍列表,每个元素是一个字典。 :param isbn: 要修改的书籍的 ISBN。 :return: 修改后的书籍列表。 """ for idx, entry in enumerate(books): if entry['ISBN'] == str(isbn): books[idx]['ISBN'] = '666' return books
代码解释:
立即学习“Python免费学习笔记(深入)”;
- for idx, entry in enumerate(books):: 遍历书籍列表,idx 是索引,entry 是书籍字典。
- if entry[‘ISBN’] == str(isbn):: 检查当前书籍的 ISBN 是否与给定的 ISBN 匹配。注意将 isbn 转换为字符串类型进行比较。
- books[idx][‘ISBN’] = ‘666’: 如果找到匹配的书籍,则将其 ISBN 的值修改为 ‘666’。
- return books: 返回修改后的书籍列表。
3. writeBooks(books, file):将数据写回文件
此函数接收书籍列表和文件名作为输入,并将书籍信息以原始格式写回文件。
def writeBooks(books, file): """ 将书籍列表以原始格式写回文件。 :param books: 书籍列表,每个元素是一个字典。 :param file: 要写入的文件名。 """ with open(file, 'w') as booksFile: for entry in books: booksFile.writelines(' -- '.join(['='.join([k,v]) for k,v in entry.items()]) + 'n')
代码解释:
立即学习“Python免费学习笔记(深入)”;
- with open(file, ‘w’) as booksFile:: 以写入模式打开文件。
- for entry in books:: 遍历书籍列表。
- ‘ — ‘.join([‘=’.join([k,v]) for k,v in entry.items()]): 将书籍字典转换为原始格式的字符串。首先,[‘=’.join([k,v]) for k,v in entry.items()] 将每个键值对转换为 “key=value” 格式的字符串,然后 ‘ — ‘.join(…) 将这些字符串用 ” — ” 连接起来。
- booksFile.writelines(… + ‘n’): 将书籍信息写入文件,并在末尾添加换行符。
使用示例
# 读取书籍信息 books = getBooks('books.txt') # 修改 ISBN 为 33 的书籍的 ISBN 值 books = decrementBookUnits(books, 33) # 将修改后的书籍信息写回文件 writeBooks(books, 'books2.txt')
注意事项
- 确保 books.txt 文件存在,并且格式正确。
- writeBooks 函数会将修改后的数据写入到 books2.txt 文件中,而不是覆盖原始文件。如果需要覆盖原始文件,请将 writeBooks 函数的第二个参数修改为 ‘books.txt’。
- 在实际应用中,可以根据需要修改 decrementBookUnits 函数中的修改逻辑,例如,可以将 ISBN 减 1,而不是直接设置为 ‘666’。
- 异常处理:在实际应用中,应该添加适当的异常处理机制,例如,处理文件不存在、文件格式错误等情况。
总结
通过将文件读取、数据处理和文件写入分离成独立的函数,可以有效提高代码的可读性和可重用性。这种模块化的设计方法使得代码更易于维护和扩展。同时,使用适当的数据结构(如字典列表)可以简化数据处理过程。
评论(已关闭)
评论已关闭