缩短获取路径的循环

Question

我目前正在尝试获取多个目录中txt文件的完整路径名。
目录结构看起来像这样，其中包含数千个图像文件夹：

• root_dir
  • image1
    • image1.gt.txt
    • image1.stuff.png
    • image1.stuff2.png
  • image2
    • image2.gt.txt
    • 等

当前代码为：

for root, dirs, files in os.walk(pathname, topdown=False):
    for file in files:
        if file.split('.')[-1] == 'txt':
            gt_txt = file
            full_path = os.path.join(root, gt_txt)

是否有办法在代码和/或运行时方面缩短此循环，或者缩短循环是否合理？