แยกส่วนขยายจากชื่อไฟล์ใน Python

มีฟังก์ชั่นในการดึงนามสกุลจากชื่อไฟล์หรือไม่?

ใช่. ใช้os.path.splitext(ดูเอกสาร Python 2.XหรือเอกสารPython 3.X ):

>>> import os
>>> filename, file_extension = os.path.splitext('/path/to/somefile.ext')
>>> filename
'/path/to/somefile'
>>> file_extension
'.ext'

ไม่เหมือนกับความพยายามในการแยกสตริงด้วยตนเองส่วนใหญ่os.path.splitextจะถือว่า/a/b.c/dเป็นไม่มีส่วนขยายอย่างถูกต้องแทนที่จะมีส่วนขยาย.c/dและจะถือว่า.bashrcไม่มีส่วนขยายแทนที่จะมีส่วนขยาย.bashrc:

>>> os.path.splitext('/a/b.c/d')
('/a/b.c/d', '')
>>> os.path.splitext('.bashrc')
('.bashrc', '')

import os.path
extension = os.path.splitext(filename)[1]

ใหม่ในเวอร์ชัน 3.4

import pathlib

print(pathlib.Path('yourPath.example').suffix) # '.example'

ฉันประหลาดใจที่ยังไม่มีใครพูดถึงpathlibมันpathlibเยี่ยมยอด!

หากคุณต้องการคำต่อท้ายทั้งหมด (เช่นถ้าคุณมี.tar.gz) .suffixesจะส่งคืนรายการของพวกเขา!

import os.path
extension = os.path.splitext(filename)[1][1:]

เพื่อให้ได้ข้อความของส่วนขยายโดยไม่มีจุด

ตัวเลือกหนึ่งอาจแยกออกจากจุด:

>>> filename = "example.jpeg"
>>> filename.split(".")[-1]
'jpeg'

ไม่มีข้อผิดพลาดเมื่อไฟล์ไม่มีส่วนขยาย:

>>> "filename".split(".")[-1]
'filename'

แต่คุณต้องระวัง:

>>> "png".split(".")[-1]
'png'    # But file doesn't have an extension

คุณควรคิดว่าทำไม JPG ถึงไม่ปรากฏในรายการของคุณ

os.path.splitext(filename)[1][1:].strip().lower()

วิธีการแก้ปัญหาใด ๆ ข้างต้นทำงาน แต่ใน linux ฉันได้พบว่ามีการขึ้นบรรทัดใหม่ในตอนท้ายของสตริงส่วนขยายซึ่งจะป้องกันไม่ให้ตรงกับการประสบความสำเร็จ เพิ่มstrip()วิธีการไปยังจุดสิ้นสุด ตัวอย่างเช่น:

import os.path
extension = os.path.splitext(filename)[1][1:].strip() 

ด้วย splitext มีปัญหากับไฟล์ที่มีนามสกุลคู่ (เช่นfile.tar.gz, file.tar.bz2ฯลฯ .. )

>>> fileName, fileExtension = os.path.splitext('/path/to/somefile.tar.gz')
>>> fileExtension 
'.gz'

แต่ควรจะ: .tar.gz

ทางออกที่เป็นไปได้อยู่ที่นี่

คุณสามารถค้นหาสิ่งที่ยอดเยี่ยมบางอย่างในโมดูล pathlib (มีให้ใน python 3.x)

import pathlib
x = pathlib.PurePosixPath("C:\\Path\\To\\File\\myfile.txt").suffix
print(x)

# Output 
'.txt'

แม้ว่าจะเป็นหัวข้อเก่า แต่ฉันสงสัยว่าทำไมไม่มีใครพูดถึง api ง่าย ๆ ของงูใหญ่ที่เรียกว่า rpartition ในกรณีนี้:

ในการรับส่วนขยายของไฟล์ที่ระบุพา ธ สัมบูรณ์คุณสามารถพิมพ์:

filepath.rpartition('.')[-1]

ตัวอย่าง:

path = '/home/jersey/remote/data/test.csv'
print path.rpartition('.')[-1]

จะให้คุณ: 'csv'

เพียงแค่ทุกคนjoinpathlib suffixes

>>> x = 'file/path/archive.tar.gz'
>>> y = 'file/path/text.txt'
>>> ''.join(pathlib.Path(x).suffixes)
'.tar.gz'
>>> ''.join(pathlib.Path(y).suffixes)
'.txt'

แปลกใจที่สิ่งนี้ยังไม่ได้กล่าวถึง:

import os
fn = '/some/path/a.tar.gz'

basename = os.path.basename(fn)  # os independent
Out[] a.tar.gz

base = basename.split('.')[0]
Out[] a

ext = '.'.join(basename.split('.')[1:])   # <-- main part

# if you want a leading '.', and if no result `None`:
ext = '.' + ext if ext else None
Out[] .tar.gz

ประโยชน์ที่ได้รับ:

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

ฟังก์ชั่น:

def get_extension(filename):
    basename = os.path.basename(filename)  # os independent
    ext = '.'.join(basename.split('.')[1:])
    return '.' + ext if ext else None

คุณสามารถใช้splita filename:

f_extns = filename.split(".")
print ("The extension of the file is : " + repr(f_extns[-1]))

สิ่งนี้ไม่ต้องการไลบรารีเพิ่มเติม

filename='ext.tar.gz'
extension = filename[filename.rfind('.'):]

นี่เป็นเทคนิคการนำเสนอสตริงโดยตรง: ฉันเห็นวิธีแก้ปัญหามากมายที่กล่าวถึง แต่ฉันคิดว่าส่วนใหญ่มองแยก แยกได้อย่างไรก็ตามไม่ว่าจะเกิดขึ้นที่ "." ทุกครั้ง . สิ่งที่คุณต้องการจะมองหาคือพาร์ติชัน

string = "folder/to_path/filename.ext"
extension = string.rpartition(".")[-1]

โซลูชันอื่นที่มีการแยกขวา:

# to get extension only

s = 'test.ext'

if '.' in s: ext = s.rsplit('.', 1)[1]

# or, to get file name and extension

def split_filepath(s):
    """
    get filename and extension from filepath 
    filepath -> (filename, extension)
    """
    if not '.' in s: return (s, '')
    r = s.rsplit('.', 1)
    return (r[0], r[1])

แม้แต่คำถามนี้ก็ได้รับคำตอบแล้วฉันก็จะเพิ่มโซลูชันใน Regex

>>> import re
>>> file_suffix = ".*(\..*)"
>>> result = re.search(file_suffix, "somefile.ext")
>>> result.group(1)
'.ext'

ซับในตัวจริงถ้าคุณชอบ regex และมันก็ไม่สำคัญว่าคุณจะมี "." เพิ่มเติม ระหว่างกลาง

import re

file_ext = re.search(r"\.([^.]+)$", filename).group(1)

ดูผลลัพธ์ที่นี่ : คลิกที่นี่

นี่คือวิธีที่ง่ายที่สุดที่จะได้รับทั้งชื่อไฟล์และส่วนต่อขยายในเพียงเส้นเดียว

fName, ext = 'C:/folder name/Flower.jpeg'.split('/')[-1].split('.')

>>> print(fName)
Flower
>>> print(ext)
jpeg

ไม่เหมือนกับโซลูชันอื่น ๆ คุณไม่จำเป็นต้องนำเข้าแพ็คเกจใด ๆ

สำหรับ funsies ... เพียงรวบรวมนามสกุลใน dict และติดตามทั้งหมดในโฟลเดอร์ จากนั้นดึงส่วนขยายที่คุณต้องการ

import os

search = {}

for f in os.listdir(os.getcwd()):
    fn, fe = os.path.splitext(f)
    try:
        search[fe].append(f)
    except:
        search[fe]=[f,]

extensions = ('.png','.jpg')
for ex in extensions:
    found = search.get(ex,'')
    if found:
        print(found)

ลองนี้:

files = ['file.jpeg','file.tar.gz','file.png','file.foo.bar','file.etc']
pen_ext = ['foo', 'tar', 'bar', 'etc']

for file in files: #1
    if (file.split(".")[-2] in pen_ext): #2
        ext =  file.split(".")[-2]+"."+file.split(".")[-1]#3
    else:
        ext = file.split(".")[-1] #4
    print (ext) #5

1. รับชื่อไฟล์ทั้งหมดที่อยู่ในรายการ
2. แยกชื่อไฟล์และตรวจสอบส่วนขยายสุดท้ายแล้วมันอยู่ในรายการ pen_ext หรือไม่
3. ถ้าใช่แล้วเข้าร่วมกับนามสกุลสุดท้ายและตั้งเป็นนามสกุลของไฟล์
4. ถ้าไม่ใช่ให้ใส่นามสกุลสุดท้ายเป็นนามสกุลไฟล์
5. จากนั้นตรวจสอบ

# try this, it works for anything, any length of extension
# e.g www.google.com/downloads/file1.gz.rs -> .gz.rs

import os.path

class LinkChecker:

    @staticmethod
    def get_link_extension(link: str)->str:
        if link is None or link == "":
            return ""
        else:
            paths = os.path.splitext(link)
            ext = paths[1]
            new_link = paths[0]
            if ext != "":
                return LinkChecker.get_link_extension(new_link) + ext
            else:
                return ""

def NewFileName(fichier):
    cpt = 0
    fic , *ext =  fichier.split('.')
    ext = '.'.join(ext)
    while os.path.isfile(fichier):
        cpt += 1
        fichier = '{0}-({1}).{2}'.format(fic, cpt, ext)
    return fichier

name_only=file_name[:filename.index(".")

ซึ่งจะทำให้ชื่อไฟล์ของคุณเป็น "." อันแรกซึ่งจะเป็นชื่อสามัญที่สุด