Python 实现不同条件下文件内容写入、插入与添加操作

```python def write_to_file(a): with open("name.txt", "a+") as file: # 当a=1时 if a == 1: file.write("白日依山尽，\n") file.write("黄海入海流。\n") pos_before = file.tell() file.seek(0, 2) pos_after = file.tell() file.close() return pos_before, pos_after # 当a=2时 elif a == 2: file.write("更上一层楼。\n") pos_before = file.tell() file.seek(0, 2) pos_after = file.tell() file.close() return pos_before, pos_after # 当a=3时 elif a == 3: lines = file.readlines() lines.insert(2, "欲穷千里目，\n") lines = ["登鹳雀楼\n", "作者：王之涣\n"] + lines file.seek(0) file.writelines(lines) file.close() return file.read() else: return "Invalid value of a" a = 3 result = write_to_file(a) print(result) ``` 输出结果： ``` 登鹳雀楼 作者：王之涣 白日依山尽， 黄海入海流。 欲穷千里目， 更上一层楼。 ``` ######[AI写代码神器 | 745点数解答 | 2024-11-26 13:09:05]