python - 如何将文本导出到新文件，userinput？

Question

所以我用 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()

Answer 1

试试这个:

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