函数更改不同扩展类型的名称文件

Question

我正在编写一个python脚本，在文件末尾替换并添加一个命令。我正在使用：

outfilename=infile.replace(".pdb","-solv"+str(solvcycle)
            + ".pdb")

是否可以让这个函数替换.pdb或.xyz文件，或者更一般地，替换infile的任何扩展名？我需要脚本替换任何infile的名称，就像它对pdb所做的那样

Answer 1

我建议使用pathlib：

from pathlib import Path

infile = 'foo/bar.xyz'
infile = Path(infile)
new_name = infile.stem + '-solv' + infile.suffix
outfilename = infile.with_name(new_name)
print( str(outfilename) )
# output: foo/bar-solv.xyz