Python中是否有一个内置函数可以替换(或删除)文件名的扩展名(如果有的话)?

例子:

print replace_extension('/home/user/somefile.txt', '.jpg')

在我的例子中,/home/user/somefile.txt将变成/home/user/somefile.jpg

我不知道这是否重要,但我需要这个我正在编写的SCons模块。(所以也许有一些特定的SCons函数我可以使用?)

我想要干净的。对字符串中出现的.txt进行简单的字符串替换显然是不干净的。(这将失败,如果我的文件名是somefile.txt.txt)


尝试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')

正如@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索引到它中,以获得文件名减去扩展名。


另一种方法是使用str.rpartition(sep)方法。

例如:

filename = '/home/user/somefile.txt'
(prefix, sep, suffix) = filename.rpartition('.')

new_filename = prefix + '.jpg'

print new_filename

我更喜欢下面使用str.rsplit()的一行程序方法:

my_filename.rsplit('.', 1)[0] + '.jpg'

例子:

>>> my_filename = '/home/user/somefile.txt'
>>> my_filename.rsplit('.', 1)
>>> ['/home/user/somefile', 'txt']

对于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'

扩展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

处理多个扩展

在使用pathlib和str.replace有多个扩展的情况下:

删除/条扩展

>>> from pathlib import Path
>>> p = Path("/path/to/myfile.tar.gz")
>>> extensions = "".join(p.suffixes)

# any python version
>>> str(p).replace(extensions, "")
'/path/to/myfile'

# python>=3.9
>>> str(p).removesuffix(extensions)
'/path/to/myfile'

替换扩展

>>> p = Path("/path/to/myfile.tar.gz")
>>> extensions = "".join(p.suffixes)
>>> new_ext = ".jpg"
>>> str(p).replace(extensions, new_ext)
'/path/to/myfile.jpg'

如果你还想要一个pathlib对象输出,那么显然你可以在Path()中换行

>>> Path(str(p).replace("".join(p.suffixes), ""))
PosixPath('/path/to/myfile')

把它们都包装在一个函数中

from pathlib import Path
from typing import Union

PathLike = Union[str, Path]


def replace_ext(path: PathLike, new_ext: str = "") -> Path:
    extensions = "".join(Path(path).suffixes)
    return Path(str(p).replace(extensions, new_ext))


p = Path("/path/to/myfile.tar.gz")
new_ext = ".jpg"

assert replace_ext(p, new_ext) == Path("/path/to/myfile.jpg")
assert replace_ext(str(p), new_ext) == Path("/path/to/myfile.jpg")
assert replace_ext(p) == Path("/path/to/myfile")
    

TLDR: 在我看来,替换所有扩展的最佳方法如下。

import pathlib
p = pathlib.Path('/path/to.my/file.foo.bar.baz.quz')
print(p.with_name(p.name.split('.')[0]).with_suffix('.jpg'))

再答: 最好的方法取决于你的python版本以及你需要处理多少扩展。也就是说,我很惊讶没有人提到pathlib的with_name。我还担心这里的一些答案不处理。在父目录中。以下是完成扩展替换的几种方法。

使用路径对象

更换最多一个分机

import pathlib
p = pathlib.Path('/path/to.my/file.foo')
print(p.with_suffix('.jpg'))

最多更换两个分机

import pathlib
p = pathlib.Path('/path/to.my/file.foo.bar')
print(p.with_name(p.stem).with_suffix('.jpg'))

更换所有扩展

使用pathlibs with_name(在我看来,最好的解决方案):

import pathlib
p = pathlib.Path('/path/to.my/file.foo.bar.baz.quz')
print(p.with_name(p.name.split('.')[0]).with_suffix('.jpg'))

使用functools。Reduce和pathlib的with_suffix:

import pathlib
import functools
p = pathlib.Path('/path/to.my/file.foo.bar.baz.quz')
print(functools.reduce(lambda v, _: v.with_suffix(''), p.suffixes, p).with_suffix('.jpg'))
print(functools.reduce(lambda v, e: v.with_suffix(e), ['' for _ in p.suffixes] + ['.jpg'], p))

Python 3.9+使用pathlib和str.removesuffix:

import pathlib
p = pathlib.Path('/path/to.my/file.foo.bar.baz.quz')
print(pathlib.Path(str(p).removesuffix(''.join(p.suffixes))).with_suffix('.jpg'))

不使用路径对象(仅限字符串)

一般来说,我认为使用pathlib的解决方案更简洁,但不是每个人都能做到这一点。如果你仍在使用python2,我很抱歉。如果你没有python2的pathlib包,我真的很抱歉。

更换所有扩展

使用os.path兼容Python 2.7:

import os
ps = '/path/to.my/file.foo.bar.baz.quz'
print(os.path.join(os.path.dirname(ps), os.path.basename(ps).split('.')[0] + '.jpg'))

Python 3.9+使用removesuffix和os。Path(如果你有python3.9,为什么不使用pathlib?)

import os
ps = '/path/to.my/file.foo.bar.baz.quz'
print(ps.removesuffix(os.path.splitext(ps)[-1].split('.', 1)[-1]) + 'jpg')