You need to enable JavaScript to run this app.
最新活动
大模型
产品
解决方案
定价
生态与合作
支持与服务
开发者
了解我们

Python:如何统计列表中特定文件扩展名项数并存储到字典?

解决方法:统计文件扩展名数量并存储到字典

没问题!我来帮你搞定这个文件扩展名统计的需求,下面给你两种实用的实现方式,适配不同场景:

方法一:基础循环实现(无需额外依赖)

如果不想引入额外模块,用基础的字典操作就能完成,而且用os.path.splitext可以安全处理各种文件名边缘情况(比如带多个点的文件):

import os

# 替换成你的文件列表
file_list = ["report.pdf", "notes.txt", "image.png", "data.txt", "resume.pdf", "archive.tar.gz", "readme"]

# 初始化统计字典
ext_count = {}
# 定义你要统计的目标扩展名(带点,比如".pdf")
target_exts = [".pdf", ".txt", ".png"]

for file in file_list:
    # 提取扩展名:splitext返回(文件名主体, 扩展名)的元组
    _, ext = os.path.splitext(file)
    # 只统计目标列表里的扩展名
    if ext in target_exts:
        # 用get方法避免KeyError,首次出现时默认计数0
        ext_count[ext] = ext_count.get(ext, 0) + 1

print(ext_count)
# 预期输出: {'.pdf': 2, '.txt': 2, '.png': 1}

方法二:用Counter简化代码(更简洁)

如果你喜欢更紧凑的写法,Python内置的collections.Counter可以帮你一键完成计数,代码量更少:

import os
from collections import Counter

file_list = ["report.pdf", "notes.txt", "image.png", "data.txt", "resume.pdf", "archive.tar.gz", "readme"]
target_exts = [".pdf", ".txt", ".png"]

# 用列表推导式过滤并提取目标扩展名
ext_list = [os.path.splitext(file)[1] for file in file_list if os.path.splitext(file)[1] in target_exts]
# 用Counter统计后转成字典
ext_count = dict(Counter(ext_list))

print(ext_count)
# 预期输出和上面一致: {'.pdf': 2, '.txt': 2, '.png': 1}

处理特殊情况:复合扩展名

如果你的列表里有像archive.tar.gz这种复合扩展名,想要统计.tar.gz而不是.gz,可以用rsplit来截取最后两个后缀:

def get_complex_ext(file_name):
    parts = file_name.rsplit('.', 2)
    if len(parts) >= 2:
        return f".{parts[-2]}.{parts[-1]}"
    # 没有有效扩展名时返回空字符串
    return ""

# 示例使用
file = "archive.tar.gz"
print(get_complex_ext(file))  # 输出: .tar.gz

把这个函数融入到之前的统计逻辑里,就能实现复合扩展名的统计啦。

内容的提问来源于stack exchange,提问作者Hannan

火山引擎 最新活动