如何将文本导出到新文件，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')

Answer 2

我认为您可能只想用新名称保存文件，此python中的fileName提取扩展名谈论分解扩展名，这样您就可以添加自己的扩展名

您最终会得到

之类的东西

name, ext = os.path.splitext(userinput)
saveFile = open(name + '.txt', 'w')

Answer 3

您可能只需要更改输出文件的扩展名即可。这是将输出文件扩展名为.txt的解决方案；如果输入文件也为 .txt，则会有一个问题，但是对于输入文件的所有其他扩展程序，都应该有效。

import csv
import os
file_name = input('Name of file:')
# https://docs.python.org/3/library/os.path.html#os.path.splitext
# https://stackoverflow.com/questions/541390/extracting-extension-from-filename-in-python
file_name, file_ext_r = os.path.splitext(file_name)
file_ext_w = '.txt'
file_name_r = ''.format(file_name, file_ext_r)
file_name_w = ''.format(file_name, file_ext_w)
print('File to read:', file_name_r)
print('File to write:', file_name_w)
with open(file_name_r, 'r') as fr, open(file_name_w, 'w') as fw:
    reader = csv.reader(fr)
    for i, row in enumerate(reader):
        print(row[2])
        if i >= 2:
            fw.write(row[2] + ' ""n')

我还简化了您的逻辑，以避免将前2行写入输出文件；无需再次读取输出文件。

这个适合您？