有没有办法在使用熊猫 df.to_csv 时自动添加文件扩展名？

如何解决有没有办法在使用熊猫 df.to_csv 时自动添加文件扩展名？

我知道如何使用“defaultextension”和“filetypes”，如下：

self.filetypes = (('CSV files','*.csv'),('CSV files','*.csv'))
self.result_file = fd.asksaveasfile(filetypes = self.filetypes,defaultextension = 'csv')

我可以在输入文件名时简单地添加扩展名，但我不想这样做。如果我输入“result”作为我的文件名，我希望实际的文件名是 result.csv。

当我在做的时候，我知道我的文件类型规范看起来有点奇怪，有两个相同的选项。读取文件时，我不知道如何只提供一个选项而不会收到错误消息。这似乎有效，至少在阅读时是这样。不确定这是否是我写作时的问题。

解决方法

您可以使用 pathlib 库中的 Pathlib.path.suffix 属性。

假设您在同一目录中有一个名为“test.csv”的文件。

from pathlib import Path 
file = Path('test.csv')
df = pd.read_csv(file)
#do stuff
df.to_csv(f"new_name.{file.suffix}")
print(file.suffix)
'.csv'

,

这是我自己才发现的。我认为与其删除问题，不如提供答案，以防其他人从中受益。

我使用了 'asksaveasfile()' 函数，它实际上打开一个文件进行写入，必要时创建它。

我发现“asksaveasfilename()”会返回用户建议打开的文件名，但实际上并未打开该文件。这使我可以在打开文件并写入文件之前轻松添加我喜欢的任何扩展名。