Python 如何读取文件夹结构并将其分配给数据结构？

我只是从python开始，并试图实现以下目标：

我有一个文件夹结构（简化）：

我想将文件名读入某种数据结构中，这样可以区分哪些文件来自同一文件夹。我在一个文件夹的案例中使用了glob，但是否可以使用glob获得以下数据结构

files = [{file1, folder1}, {file2, folder1}, {file3, folder2}...]

---

我想你更喜欢这种结构：

files = {folder1: [file1, file2], folder2: [file3], ...}

以下代码将完成此操作：

import os

rootDir = '.'
files = {}
for dirName, subdirList, fileList in os.walk(rootDir):
    files[dirName] = fileList

---

|谢谢，我想那会有用的。我需要对这些文件做的是，列出2个文件的所有组合，对于每一对，我需要知道它们是否来自同一个文件夹。

import os

rootDir = '.'
files = {}
for dirName, subdirList, fileList in os.walk(rootDir):
    files[dirName] = fileList