解码输入值后创建新文件名答案

【问题标题】：Create a new file name after decoding an input value解码输入值后创建新文件名
【发布时间】：2020-08-08 10:44:59
【问题描述】：

在这篇文章中，我询问了如何使用数字作为索引来选择目录中的文件。现在我想以自动方式保存这个文件，而不是手动添加名称。例如：

我有以下列表，由代码生成（由 Eric M 建议）：

files = [f for f in glob.glob("*.txt")]

for fi, f in enumerate(files):
    print(fi, f)

query = input("Please add your selection: ") # just the number
df = pd.read_csv(files[int(query)])

输出：

1     text1.txt
2     text2.txt
3     text3.txt
...

在对数据框进行一些更改后，我想将新表保存到一个新文件中，例如

text1_test.txt

或

text2_test.txt

但是，要做到这一点，我需要取回有关我在上面代码中选择的号码的信息，即query。

我该怎么做？

【问题讨论】：

标签： python pandas

【解决方案1】：

这个解决方案可以有很多。当您读取数据框时，将文件名保存为索引名，稍后您可以使用此变量以防query 不再可用：

df = pd.read_csv(files[int(query)])
df.index.name = files[int(query)]

# Do other stuff

csv_name = df.index.name.split('.')[0] + '_test.txt'
df.to_csv(csv_name)

【讨论】：