优化代码 filename input ("请输入文件名:\ n ") fp = open ( filename , W ch = input ("请输入字符串:\ n ") while ch !='@': if '@' in ch : t = ch . find ("@") fp . write ( ch [0:t]) break else : fp . write ( ch +"\ n ") ch = input (") fp . close (
时间: 2024-01-08 10:03:02 浏览: 52
详细讲解Python中的文件I/O操作
以下是优化后的代码:
filename = input("请输入文件名:\n")
with open(filename, "w") as fp:
ch = input("请输入字符串:\n")
while '@' not in ch:
fp.write(ch + "\n")
ch = input()
t = ch.find("@")
fp.write(ch[:t])
这里使用了 with 语句来自动关闭文件,同时也将写文件操作放在了 with 语句内部,可以更好地保证文件的正常关闭。另外,将判断 '@' 是否在输入字符串中的条件改为了 while '@' not in ch,这样更加简洁易懂。最后,将查找 '@' 的操作放在了循环外面,避免了重复查找。
阅读全文