注册公司一年大概需要多少费用搜索引擎优化培训班
可以使用 Python 的 os
模块或者 pathlib
模块来获取指定路径下所有图片的名称。以下是使用这两种方法实现的代码示例:
使用 os
模块
import osdef get_image_names_os(path):image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp')image_names = []for root, dirs, files in os.walk(path):for file in files:if file.lower().endswith(image_extensions):image_names.append(file)return image_names# 示例用法
path = '.' # 当前目录
image_names = get_image_names_os(path)
print(image_names)
使用 pathlib
模块
from pathlib import Pathdef get_image_names_pathlib(path):image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp')path = Path(path)image_names = [p.name for p in path.rglob('*') if p.is_file() and p.suffix.lower() in image_extensions]return image_names# 示例用法
path = '.' # 当前目录
image_names = get_image_names_pathlib(path)
print(image_names)
上述代码定义了两个函数 get_image_names_os
和 get_image_names_pathlib
分别使用 os
模块和 pathlib
模块来获取指定路径下所有图片的名称。函数接受一个路径作为参数,返回一个包含所有图片名称的列表。你可以根据需要修改 path
变量的值来指定不同的路径。