python 获取文件夹图片名称后写进列表

Python语言提供了os模块和glob模块，可以用来获取文件夹中的所有文件，这些文件名可以装入列表中。下面是一个Python程序示例，演示如何获取文件夹中的图片文件：

import os
import glob

#定义函数，获取文件夹中的所有图片文件，存入列表中
def get_images(folder_path):
    #先用os模块获取文件夹中的所有文件列表
    files = os.listdir(folder_path)
    #然后用glob模块过滤出所有图片文件
    images = [f for f in files if f.endswith('.jpg') or f.endswith('.jpeg') or f.endswith('.png')]
    return images

#测试一下，列出文件夹中所有图片文件的文件名
folder_path = 'D:/Python/image_folder'
image_list = get_images(folder_path)
print(image_list)

这个程序中，我们首先定义了一个函数`get_images()`，它接受一个参数`folder_path`，表示文件夹路径。在函数内部，我们首先使用`os.listdir()`函数获取文件夹中的所有文件列表，然后使用一个列表推导式和`str.endswith()`方法对所有以`.jpg`、`.jpeg`、`.png`结尾的文件进行过滤，最终得到图片文件列表`images`。函数返回这个列表。

在程序的后面部分，我们定义了一个测试变量`folder_path`，赋值为一个文件夹路径，然后调用`get_images()`函数得到文件夹中所有图片文件的文件名列表，最后打印输出这个列表。这个程序演示了如何利用Python语言快速、高效地获取文件夹中的所有图片文件名，可以方便地用于批量处理图片等应用场景。