python - 用户输入如何将文本导出到新文件？

所以我用python写了一个小程序，它允许我提取一个.csv文件，过滤掉我需要的行，然后将它们导出到一个新的.txt文件中。
这工作得很好，所以我决定通过允许用户选择应该由他自己通过控制台（命令行）转换的文件来使其更加用户友好。

我的问题：该文件导入为.csv文件，但未导出为.txt文件，这导致我的程序覆盖了原始文件，由于程序中的某个步骤，该文件将被清空，这使我可以删除前两行输出文本。

有人知道解决方案吗？

谢谢 ：）

import csv
import sys

userinput = raw_input('List:')
saveFile = open(userinput, 'w')

with open(userinput, 'r') as file:
    reader = csv.reader(file)
    count = 0

    for row in reader:
        print(row[2])
        saveFile.write(row[2] + ' ""\n')
saveFile.close()

saveFile = open(userinput, 'r')
data_list = saveFile.readlines()
saveFile.close()

del data_list[1:2]

saveFile = open(userinput, 'w')
saveFile.writelines(data_list)
saveFile.close()

最佳答案

尝试这个：

userinput = raw_input('List:')
f_extns = userinput.split(".")
saveFile = open(f_extns[0]+'.txt', 'w')

关于python - 用户输入如何将文本导出到新文件？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/55288103/