Python中是否有一个内置函数可以替换(或删除)文件名的扩展名(如果有的话)?
例子:
print replace_extension('/home/user/somefile.txt', '.jpg')
在我的例子中，/home/user/somefile.txt将变成/home/user/somefile.jpg
我不知道这是否重要，但我需要这个我正在编写的SCons模块。(所以也许有一些特定的SCons函数我可以使用?)
我想要干净的。对字符串中出现的.txt进行简单的字符串替换显然是不干净的。(这将失败，如果我的文件名是somefile.txt.txt)
当前回答
扩展AnaPana的答案，如何使用pathlib删除一个扩展(Python >= 3.4):
>>> from pathlib import Path
>>> filename = Path('/some/path/somefile.txt')
>>> filename_wo_ext = filename.with_suffix('')
>>> filename_replace_ext = filename.with_suffix('.jpg')
>>> print(filename)
/some/path/somefile.ext
>>> print(filename_wo_ext)
/some/path/somefile
>>> print(filename_replace_ext)
/some/path/somefile.jpg
其他回答
我更喜欢下面使用str.rsplit()的一行程序方法:
my_filename.rsplit('.', 1)[0] + '.jpg'
例子:
>>> my_filename = '/home/user/somefile.txt'
>>> my_filename.rsplit('.', 1)
>>> ['/home/user/somefile', 'txt']
另一种方法是使用str.rpartition(sep)方法。
例如:
filename = '/home/user/somefile.txt'
(prefix, sep, suffix) = filename.rpartition('.')
new_filename = prefix + '.jpg'
print new_filename
正如@jethro所说，splitext是一种简洁的方法。但在这种情况下，自己拆分它很容易，因为扩展名必须是文件名中最后一个句号之后的部分:
filename = '/home/user/somefile.txt'
print( filename.rsplit( ".", 1 )[ 0 ] )
# '/home/user/somefile'
rsplit告诉Python从字符串的右侧开始执行字符串拆分，而1表示最多执行一次拆分(因此例如:“foo.bar。Baz ' -> ['foo。Bar '， 'baz'])。由于rsplit总是返回一个非空数组，我们可以安全地将0索引到它中，以获得文件名减去扩展名。
对于Python >= 3.4:
from pathlib import Path
filename = '/home/user/somefile.txt'
p = Path(filename)
new_filename = p.parent.joinpath(p.stem + '.jpg') # PosixPath('/home/user/somefile.jpg')
new_filename_str = str(new_filename) # '/home/user/somefile.jpg'
尝试os.path.splitext，它应该做你想要的。
import os
print os.path.splitext('/home/user/somefile.txt')[0]+'.jpg' # /home/user/somefile.jpg
os.path.splitext('/home/user/somefile.txt') # returns ('/home/user/somefile', '.txt')
