Có một hàm tích hợp sẵn trong Python sẽ thay thế (hoặc loại bỏ, bất cứ điều gì) phần mở rộng của tên tệp (nếu nó có)?

Thí dụ:

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

Trong ví dụ của tôi: /home/user/somefile.txtsẽ trở thành/home/user/somefile.jpg

Tôi không biết nó có quan trọng không, nhưng tôi cần cái này cho một mô-đun SCons mà tôi đang viết. (Vì vậy, có lẽ có một số chức năng SCons cụ thể mà tôi có thể sử dụng?)

Tôi muốn một cái gì đó sạch sẽ . Thực hiện thay thế chuỗi đơn giản của tất cả các lần xuất hiện .txttrong chuỗi rõ ràng là không sạch. (Điều này sẽ không thành công nếu tên tệp của tôi là somefile.txt.txt.txt)

Hãy thử os.path.splitext nó sẽ làm những gì bạn muốn.

import os
print os.path.splitext('/home/user/somefile.txt')[0]+'.jpg'

Mở rộng câu trả lời của AnaPana, cách xóa tiện ích mở rộng bằng cách sử dụng 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

Như @jethro đã nói, đây splitextlà một cách làm gọn gàng. Nhưng trong trường hợp này, khá dễ dàng để tự tách nó ra, vì phần mở rộng phải là một phần của tên tệp sau dấu chấm cuối cùng:

filename = '/home/user/somefile.txt'
print( filename.rsplit( ".", 1 )[ 0 ] )
# '/home/user/somefile'

Lệnh rsplitcho Python thực hiện tách chuỗi bắt đầu từ bên phải của chuỗi và1 thực hiện nhiều nhất một lần tách (ví dụ: 'foo.bar.baz'-> [ 'foo.bar', 'baz' ]). Vì rsplitsẽ luôn trả về một mảng không trống, chúng tôi có thể lập chỉ mục an toàn 0vào đó để lấy tên tệp trừ đi phần mở rộng.

Tôi thích cách tiếp cận một lớp sau bằng cách sử dụng str.rsplit () :

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

Thí dụ:

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

Đối với 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'

Xử lý nhiều tiện ích mở rộng

Trong trường hợp bạn có nhiều tiện ích mở rộng, một lớp lót này sẽ sử dụng pathlibvà str.replacehoạt động:

Xóa / tách phần mở rộng

>>> from pathlib import Path
>>> p = Path("/path/to/myfile.tar.gz")
>>> str(p).replace("".join(p.suffixes), "")
'/path/to/myfile'

Thay thế phần mở rộng

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

Nếu bạn cũng muốn có một pathlibđầu ra đối tượng thì rõ ràng bạn có thể quấn dòng vàoPath()

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

Gói gọn tất cả trong một hàm

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

Một cách khác để làm là sử dụng str.rpartition(sep)phương pháp.

Ví dụ:

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

new_filename = prefix + '.jpg'

print new_filename