在Python中从文件名中提取扩展名

Question

有没有从文件名中提取扩展名的函数？

Answer 1

使用 os.path.splitext ：

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

与大多数手动字符串分割尝试不同，os.path.splitext 会正确地将 /a/bc/d 视为没有扩展名，而不是具有扩展名 < code>.c/d，它会将 .bashrc 视为没有扩展名，而不是具有扩展名 .bashrc：

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

Answer 2

3.4 版中的新功能。

import pathlib

print(pathlib.Path('yourPath.example').suffix) # '.example'
print(pathlib.Path("hello/foo.bar.tar.gz").suffixes) # ['.bar', '.tar', '.gz']
print(pathlib.Path('/foo/bar.txt').stem) # 'bar'

我很惊讶没有人提到< code>pathlib 然而，pathlib 太棒了！

Answer 3

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

Answer 4

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

仅获取扩展名的文本，不带点。

Answer 5

对于简单的用例，一个选项可能会从点中分离出来：

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

当文件没有扩展名时不会出错：

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

但你必须小心：

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

也不适用于 Unix 系统中的隐藏文件：

>>> ".bashrc".split(".")[-1]
'bashrc'    # But this is not an extension

对于一般用途，更喜欢 os.path.splitext

Answer 6

值得在其中添加一个较低的值，这样您就不会发现自己想知道为什么 JPG 没有出现在您的列表中。

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

Answer 7

上述任何解决方案都有效，但在 Linux 上我发现扩展字符串末尾有一个换行符，这将阻止匹配成功。 将 strip() 方法添加到末尾。 例如：

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

Answer 8

使用 splitext 时，具有双扩展名的文件会出现问题（例如 file.tar.gz、file.tar.bz2 等），

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

但应该是：。 tar.gz

可能的解决方案在此处

Answer 9

您可以在 pathlib 模块（在 python 3.x 中可用）中找到一些很棒的东西。

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

# Output 
'.txt'

Answer 10

只需加入所有pathlib后缀即可。

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

Answer 11

虽然这是一个老话题，但我想知道为什么没有人提到一个非常简单的 python api，称为 rpartition 在这种情况下：

要获得给定文件绝对路径的扩展名，您可以简单地输入：

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

example:

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

will get you: 'csv '

Answer 12

令人惊讶的是，还没有提到这一点：

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

优点：

• 按我能想到的任何方式工作
• 没有模块
• 没有正则表达式
• 跨平台
• 易于扩展（例如没有用于扩展的前导点，只有扩展的最后一部分）

作为功能：

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

Answer 13

您可以在文件名上使用split：

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

这不需要额外的库

Answer 14

在 Python 中从文件名中提取扩展名

Python os 模块 splitext()

splitext() 函数将文件路径拆分为具有两个值的元组 - 根和扩展名。

import os
# unpacking the tuple
file_name, file_extension = os.path.splitext("/Users/Username/abc.txt")
print(file_name)
print(file_extension)

使用Pathlib模块获取文件扩展名

Pathlib模块获取文件扩展名

import pathlib
pathlib.Path("/Users/pankaj/abc.txt").suffix
#output:'.txt'

Answer 15

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

Answer 16

即使这个问题已经得到解答，我也会在正则表达式中添加解决方案。

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

Answer 17

这是一种直接的字符串表示技术：
我看到提到了很多解决方案，但我认为大多数都在考虑拆分。
然而，Split 在每次出现“.”时都会执行此操作。 。
您更愿意寻找的是分区。

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

Answer 18

另一种右分割的解决方案：

# 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])

Answer 19

您可以使用以下代码来拆分文件名和扩展名。

    import os.path
    filenamewithext = os.path.basename(filepath)
    filename, ext = os.path.splitext(filenamewithext)
    #print file name
    print(filename)
    #print file extension
    print(ext)

Answer 20

好吧，我知道我迟到了，

这是我的简单解决方案

file = '/foo/bar/whatever.ext'
extension = file.split('.')[-1]
print(extension)

#output will be ext

Answer 21

如果您喜欢正则表达式，那么这就是真正的一句台词。
即使你有额外的“。”也没关系。 在中间

import re

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

查看结果：单击此处

Answer 22

您可以使用 endswith 来识别 python 中的文件扩展名，

如下例所示

for file in os.listdir():
    if file.endswith('.csv'):
        df1 =pd.read_csv(file)
        frames.append(df1)
        result = pd.concat(frames)

Answer 23

试试这个：

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. ，然后检查出来

Answer 24

最简单的获取方法是使用mimtypes，下面是示例：

import mimetypes

mt = mimetypes.guess_type("file name")
file_extension =  mt[0]
print(file_extension)

Answer 25

我肯定迟到了，但万一有人想在不使用另一个库的情况下实现这一目标：

file_path = "example_tar.tar.gz"
file_name, file_ext = [file_path if "." not in file_path else file_path.split(".")[0], "" if "." not in file_path else file_path[file_path.find(".") + 1:]]
print(file_name, file_ext)

第二行基本上只是以下代码，但塞进一行：

def name_and_ext(file_path):
    if "." not in file_path:
        file_name = file_path
    else:
        file_name = file_path.split(".")[0]
    if "." not in file_path:
        file_ext = ""
    else:
        file_ext = file_path[file_path.find(".") + 1:]
    return [file_name, file_ext]

即使这有效，它也可能不起作用文件类型，特别是 .zshrc，我建议使用 os 的 os.path.splitext 函数，示例如下：

import os
file_path = "example.tar.gz"
file_name, file_ext = os.path.splitext(file_path)
print(file_name, file_ext)

干杯 :)

Answer 26

对于有趣的人来说......只需将扩展名收集在字典中，然后在文件夹中跟踪所有扩展名即可。 然后只需拉出你想要的扩展即可。

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)

Answer 27

此方法需要字典、列表或集合。 您可以使用内置字符串方法来使用“.endswith”。 这将在文件末尾的列表中搜索名称，只需使用 str.endswith(fileName[index]) 即可完成。 这更多的是为了获取和比较扩展。

https://docs.python.org/3/library/stdtypes。 html#string-methods

示例 1:

dictonary = {0:".tar.gz", 1:".txt", 2:".exe", 3:".js", 4:".java", 5:".python", 6:".ruby",7:".c", 8:".bash", 9:".ps1", 10:".html", 11:".html5", 12:".css", 13:".json", 14:".abc"} 
for x in dictonary.values():
    str = "file" + x
    str.endswith(x, str.index("."), len(str))

示例 2:

set1 = {".tar.gz", ".txt", ".exe", ".js", ".java", ".python", ".ruby", ".c", ".bash", ".ps1", ".html", ".html5", ".css", ".json", ".abc"}
for x in set1:
   str = "file" + x
   str.endswith(x, str.index("."), len(str))

示例 3:

fileName = [".tar.gz", ".txt", ".exe", ".js", ".java", ".python", ".ruby", ".c", ".bash", ".ps1", ".html", ".html5", ".css", ".json", ".abc"];
for x in range(0, len(fileName)):
    str = "file" + fileName[x]
    str.endswith(fileName[x], str.index("."), len(str))

示例 4

fileName = [".tar.gz", ".txt", ".exe", ".js", ".java", ".python", ".ruby", ".c", ".bash", ".ps1", ".html", ".html5", ".css", ".json", ".abc"];
str = "file.txt"
str.endswith(fileName[1], str.index("."), len(str))

带有输出的示例 5、6、7

示例 8

fileName = [".tar.gz", ".txt", ".exe", ".js", ".java", ".python", ".ruby", ".c", ".bash", ".ps1", ".html", ".html5", ".css", ".json", ".abc"];
exts = []
str = "file.txt"
for x in range(0, len(x)):
    if str.endswith(fileName[1]) == 1:
         exts += [x]
     

Answer 28

在这里，如果您想提取最后一个文件扩展名（如果它有多个

class functions:
    def listdir(self, filepath):
        return os.listdir(filepath)
    
func = functions()

os.chdir("C:\\Users\Asus-pc\Downloads") #absolute path, change this to your directory
current_dir = os.getcwd()

for i in range(len(func.listdir(current_dir))): #i is set to numbers of files and directories on path directory
    if os.path.isfile((func.listdir(current_dir))[i]): #check if it is a file
        fileName = func.listdir(current_dir)[i] #put the current filename into a variable
        rev_fileName = fileName[::-1] #reverse the filename
        currentFileExtension = rev_fileName[:rev_fileName.index('.')][::-1] #extract from beginning until before .
        print(currentFileExtension) #output can be mp3,pdf,ini,exe, depends on the file on your absolute directory

输出为 mp3），即使只有 1 个扩展名也有效

Answer 29

# 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 ""

Answer 30

a = ".bashrc"
b = "text.txt"
extension_a = a.split(".")
extension_b = b.split(".")
print(extension_a[-1])  # bashrc
print(extension_b[-1])  # txt