将文件夹名称转换为键并将文件夹内部图像转换为数组

Question

我正在编写一个脚本，将文件夹的名称转换为字典键，每个文件夹中的图像内部将是每个键的数组值。我怎样才能实现它？

例子 Angelina 是文件夹名称，在她的文件夹中有属于她的图像。然后，此过程对所有文件夹保持相同。

idendities = {
    "Angelina": ["img1.jpg", "img2.jpg", "img4.jpg", "img5.jpg", "img6.jpg", "img7.jpg", "img10.jpg", "img11.jpg"],
    "Scarlett": ["img8.jpg", "img9.jpg", "img47.jpg", "img48.jpg", "img49.jpg", "img50.jpg", "img51.jpg"]}

Answer 1

你可以通过这样的方式来完成 -

import os

key_value_map = {}

# Change root with your root directory name
root = 'test'

dir_list = os.listdir(root)

for dir in dir_list:
    files_list = os.listdir(os.path.join(root, dir))
    jpg_files = []

    for f in files_list:
        if f.endswith(".jpg"):
            jpg_files.append(f)

    key_value_map[dir] = jpg_files


print(key_value_map)

假设您的工作目录中有两个目录，即 Angelina 和 Scarlet。