自动根据python中的文件名保存csv

Question

我不想保存 csv 文件 manually.Is 有可以自动完成的代码吗？首先我加载文件 1，将其转换为数据帧并执行一些没有任何重要意义的过程关于我的问题。这是文件：

file1="C:/Users/ozzgu/OneDrive/Desktop/ml articles/model/datasets/RC101.txt"
rc1=pd.read_csv(file1,sep="\s+")

处理完成后，我需要像下面这样保存txt文件。

vehicle1.to_csv("vehicle1_KMEANS3V_RC101.txt")

如果我加载 file1=RC102.txt，它有没有办法自动保存“vehicle_KMEANS3V_RC102.txt”。谢谢

Answer 1

您可以使用os库从路径中提取文件名，然后将文本附加到它，并将其传递给保存函数。

import os

filepath = "C:/Users/ozzgu/OneDrive/Desktop/ml articles/model/datasets/RC101.txt"
filename = os.path.basename(filepath)
new_filename = "vehicle1_KMEANS3V_" + filename
print(f"Old filename: {filename}")
print(f"New filename: {new_filename}")

输出：

Old filename: RC101.txt  
New filename: vehicle1_KMEANS3V_RC101.txt

自动根据python中的文件名保存csv

saving csv according to the name of file in python automatically

python

dataframe

txt