มีฟังก์ชันในตัวใน Python ที่จะแทนที่ (หรือลบอะไรก็ได้) ส่วนขยายของชื่อไฟล์ (ถ้ามี)?

ตัวอย่าง:

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

ในตัวอย่างของฉัน: /home/user/somefile.txtจะกลายเป็น/home/user/somefile.jpg

ฉันไม่รู้ว่ามันสำคัญหรือเปล่า แต่ฉันต้องการสิ่งนี้สำหรับโมดูล SCons ที่ฉันกำลังเขียนอยู่ (ดังนั้นอาจมีฟังก์ชันเฉพาะของ SCons ที่ฉันสามารถใช้ได้)

ฉันต้องการบางสิ่งบางอย่างที่สะอาด การเปลี่ยนสตริงอย่างง่ายของเหตุการณ์ทั้งหมด.txtภายในสตริงนั้นไม่สะอาดอย่างเห็นได้ชัด (สิ่งนี้จะล้มเหลวหากชื่อไฟล์ของฉันคือsomefile.txt.txt.txt)

ลองos.path.splitextมันควรทำตามที่คุณต้องการ

import os
print os.path.splitext('/home/user/somefile.txt')[0]+'.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

ดังที่ @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.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'

การจัดการส่วนขยายหลายรายการ

ในกรณีที่คุณมีส่วนขยายหลายรายการให้ใช้pathlibและใช้str.replaceงานได้:

ลบ / แถบส่วนขยาย

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

แทนที่ส่วนขยาย

>>> p = Path("/path/to/myfile.tar.gz")
>>> new_ext = ".jpg"
>>> str(p).replace("".join(p.suffixes), 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')

อีกวิธีหนึ่งที่ทำได้คือใช้str.rpartition(sep)วิธี

ตัวอย่างเช่น:

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

new_filename = prefix + '.jpg'

print new_filename