使用 os.listdir 在 Python 中选择随机 img

在 Python 中，要选择一个随机的图像文件，可以使用 os.listdir 函数。

代码示例

以下是如何实现这一功能的示例代码：

import os
import random

# 指定图像文件夹路径
img_folder = "/path/to/img/folder/"

# 获取这个文件夹中的所有图像文件
img_files = [f for f in os.listdir(img_folder) if f.endswith('.jpg') or f.endswith('.png')]

# 在图像文件列表中随机选择一个
selected_img = random.choice(img_files)

# 打印所选择的图像文件名
print("Selected image: " + selected_img)

运行此程序将随机选择并打印一个图像文件名。

代码解释

• os.listdir 函数将返回指定目录中的所有文件和子目录的名称列表。
• [] 内的代码是列表推导式，它将 os.listdir 的结果过滤为只包含以 .jpg 或 .png 结尾的图像文件名。
• random.choice 函数从列表中随机选择一个元素。在这种情况下，元素是图像文件名。
• 最后，程序输出所选择的图像文件名。

注意事项

在实际使用时，请将 img_folder 变量设置为您所需的图像文件夹的路径。此外，注意文件名的拼写和大小写。

结论

在 Python 中选择随机图像并不是一项难事，只需使用 os.listdir 和 random.choice 函数即可轻松实现。