如何得到路径下的绝对路径并且回传到list上

Question

如图所示 我的CBIS-DDSM下面有四种类型的文件夹分别是Mass-Test_， Mass-Training_， Calc-Test_， Calc-Training_， 如下四图

。

。

。

然后！ 在这四大类的文件夹中， 又分别夹着两个子文件夹. 才会到.dcm档案的地方。 举例！

好了， 问题来了。
我该如何分成四个不同的list里面装着Mass-Test_类的dcm 绝对路径， Mass-Training_的绝对路径， Calc-Test的绝对路径，Calc-Training_的绝对路径。
比如：

只是这边只分成train和test两个list来装路径，但概念是一样的，并且这个例子的路径没有那么多子文件夹夹在里面。
我到底要怎么忽略那两个子文件夹， 让我的dcm档案路径直接return回list里面呀？
请帮我集思广益，感谢。
希望我的描述够清楚， 可能看起来有点复杂。

Answer 1

写了个示例代码给你

import os

path = "CBIS-DDSM" # 根目录

mass_test_list = []
mass_train_list = []
calc_test_list = []
calc_train_list = []

for root, dirs, files in os.walk(path):
    for dir in dirs:
        if dir.startswith("Mass-Test_"):
            sub_path = os.path.join(root, dir)
            for sub_root, sub_dirs, sub_files in os.walk(sub_path):
                for sub_file in sub_files:
                    if sub_file.endswith(".dcm"):
                        mass_test_list.append(os.path.join(sub_root, sub_file))
        elif dir.startswith("Mass-Training_"):
            sub_path = os.path.join(root, dir)
            for sub_root, sub_dirs, sub_files in os.walk(sub_path):
                for sub_file in sub_files:
                    if sub_file.endswith(".dcm"):
                        mass_train_list.append(os.path.join(sub_root, sub_file))
        elif dir.startswith("Calc-Test_"):
            sub_path = os.path.join(root, dir)
            for sub_root, sub_dirs, sub_files in os.walk(sub_path):
                for sub_file in sub_files:
                    if sub_file.endswith(".dcm"):
                        calc_test_list.append(os.path.join(sub_root, sub_file))
        elif dir.startswith("Calc-Training_"):
            sub_path = os.path.join(root, dir)
            for sub_root, sub_dirs, sub_files in os.walk(sub_path):
                for sub_file in sub_files:
                    if sub_file.endswith(".dcm"):
                        calc_train_list.append(os.path.join(sub_root, sub_file))