我想更改下面的代码
for directory, dirs, files in os.walk(directory_1):
do_something()
for directory, dirs, files in os.walk(directory_2):
do_something()
到此代码:
for directory, dirs, files in os.walk(directory_1) + os.walk(directory_2):
do_something()
我得到了错误:
不支持+:'generator'和'generator'的操作数类型
如何在Python中连接两个生成器?
当前回答
Itertools.chain()应该做到这一点。它接受多个可迭代对象，并逐个产生结果，大致相当于:
def chain(*iterables):
for it in iterables:
for element in it:
yield element
使用的例子:
from itertools import chain
g = (c for c in 'ABC') # Dummy generator, just for example
c = chain(g, 'DEF') # Chain the generator and a string
for item in c:
print(item)
输出:
A
B
C
D
E
F
其他回答
如果你想从一个已知目录之前和之后获取文件路径列表，你可以这样做:
for r,d,f in os.walk(current_dir):
for dir in d:
if dir =='after':
after_dir = os.path.abspath(os.path.join(current_dir, dir))
for r,d,f in os.walk(after_dir):
after_flist.append([os.path.join(r,file)for file in f if file.endswith('json')])
elif dir =='before':
before_dir = os.path.abspath(os.path.join(current_dir, dir))
for r,d,f in os.walk(before_dir):
before_flist.append([os.path.join(r,file)for file in f if file.endswith('json')])
我知道有更好的答案，这是我觉得简单的代码。
Itertools.chain()应该做到这一点。它接受多个可迭代对象，并逐个产生结果，大致相当于:
def chain(*iterables):
for it in iterables:
for element in it:
yield element
使用的例子:
from itertools import chain
g = (c for c in 'ABC') # Dummy generator, just for example
c = chain(g, 'DEF') # Chain the generator and a string
for item in c:
print(item)
输出:
A
B
C
D
E
F
简单的例子:
from itertools import chain
x = iter([1,2,3]) #Create Generator Object (listiterator)
y = iter([3,4,5]) #another one
result = chain(x, y) #Chained x and y
2020年更新:在Python 3和Python 2中都可以工作
import itertools
iterA = range(10,15)
iterB = range(15,20)
iterC = range(20,25)
第一个选项
for i in itertools.chain(iterA, iterB, iterC):
print(i)
# 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
可选选项，在python 2.6中引入
for i in itertools.chain.from_iterable( [iterA, iterB, iterC] ):
print(i)
# 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24
Itertools.chain()是基本的。
Itertools.chain.from_iterable()很方便，如果你有一个包含可迭代对象的可迭代对象。例如，每个子目录的文件列表，如[["src/server.py"， "src/readme.txt"]， ["test/test.py"]]]。
您可以将任何生成器放入列表中。虽然不能组合生成器，但可以组合列表。这样做的缺点是实际上在内存中创建了3个列表，但优点是可读性非常好，不需要导入，并且是单行习惯用法。
OP解决方案。
for directory, dirs, files in list(os.walk(directory_1)) + list(os.walk(directory_2)):
do_something()
a = range(20)
b = range(10,99,3)
for v in list(a) + list(b):
print(v)
