问题描述
root/a/1.txt
root/a/2.txt
root/a/3.json
root/b/1.txt
root/b/2.txt
root/b/3.json
root/c/1.txt
root/c/2.txt
我想迭代一些特殊子目录(不是全部)中的文件。例如,获取子目录 .txt
和 a
中的所有 b
文件。
当然,我们可以通过python中的for
循环和os.listdir()/glob.glob()
轻松实现。但我正在寻找一个像 unix shell ls
一样简单的 python 包(如果存在)。
ls ./root/{a,b}/*.txt
# output:
# root/a/1.txt
# root/a/2.txt
# root/b/1.txt
# root/b/2.txt
有人知道这样的python包吗?
解决方法
从技术上讲,您可以在该示例中使用 glob,因为 a
和 b
是单个字符。
>>> glob.glob('./root/[ab]/*.txt')
['./root/a/1.txt','./root/a/2.txt','./root/b/1.txt','./root/b/2.txt']
否则你需要多个 glob 调用(你可以为其创建一个函数)
glob('./root/dir1/*.txt') + glob('./root/dir1/*.txt')
但是,您要问的是 {a,b}
,它是“shell 大括号扩展”,而不是 glob 语法的一部分。
请参阅 https://pypi.org/project/wcmatch/ 了解更多信息。
,正如上面@Karl Thornton 所说,wcmatch
包完全可以满足需求。
请在此处提供更多详细信息。
# pip install wcmatch
>>> from wcmatch import glob as wcglob
>>> wcglob.glob('./root/{a,b}/*.txt',flags=wcglob.BRACE)
wcmatch
为带有 glob
标志的 BRACE
启用 Bash 样式大括号扩展