從函數返回應用列表解析

Question

我想填充三個列表列出清單相結合。即，f返回stuff_for_A,stuff_for_B,stuff_for_C其中每個stuff是包含N個項目的列表。我想加入每個到他們各自的列表，因爲我循環迭代發送輸入到f。

如何編寫超快速列表理解來創建這3個列表？

注意：如果能讓事情變得更簡單，我可以重構f的輸出（例如將項目拉到一起）。

編輯：這是一些底部列表理解不正確的僞代碼。

def f(input): 
    x = precompute(input) 
    stuff_for_A = create_A_list(x) 
    stuff_for_B = create_B_list(x) 
    stuff_for_C = create_C_list(x) 
    return stuff_for_A,stuff_for_B,stuff_for_C 
A,B,C = [f(input) for input in iterable] 

Answer 1

直接擴展A，B，C而不返回東西？

iterable = range(10) 
A, B, C = [], [], [] 

def f(item): 
    A.extend([item * 2]) 
    B.extend([item * 3]) 
    C.extend([item * 4]) 

map(f, iterable) 
print A 
print B 
print C 

https://repl.it/Chl6

（我用.extend()裏面列出，因爲你的函數返回列表，否則使用A.add(thing)添加個性的東西。我也改名爲input到item因爲input()是Python中的內置函數）

Answer 2

看來，你想要的結果相同

A = [create_A_list(input) for input in iterable] 
B = [create_B_list(input) for input in iterable] 
C = [create_C_list(input) for input in iterable] 

但只有一個通過iterable。有了明確的for循環，你會用

A = [] 
B = [] 
C = [] 
for input in iterable: 
    A.append(create_A_input(input)) 
    B.append(create_B_input(input)) 
    C.append(create_C_input(input)) 

與一個列表理解，你可以得到三倍的列表：

triples = [ (create_A_input(input), 
      create_B_input(input), 
      create_C_input(input)) for input in iterable ] 

然後你可以轉成三重名單：

A, B, C = map(list, zip(*[(create_A_input(input), 
          create_B_input(input), 
          create_C_input(input)) for input in iterable])) 

你原來f函數返回一個三重，所以你可以使用使用到位三聯的在前面：

A, B, C = map(list, zip(*[f(input) for input in iterable)])) 

Answer 3

要創建列表從返回元組，你可以使用與zip和在發電機表達拆包多個任務：

def f(input): 
    ... 
    return stuff_for_A,stuff_for_B,stuff_for_C 

A, B, C = zip(*(f(input) for input in iterable)) 

這將返回各自的項目每個列表分組爲3元組。

---

試用：

>>> A, B, C = zip(*[(1,2,3), (4,5,6), ('x', 'y', 'z'), (8,9,0)]) 
>>> A 
(1, 4, 'x', 8) 
>>> B 
(2, 5, 'y', 9) 
>>> C 
(3, 6, 'z', 0) 

要返回他們的名單，而不是，您可以使用列表理解與以前的操作：

>>> A, B, C = [list(i) for i in zip(*[(1,2,3), (4,5,6), ('x', 'y', 'z'), (8,9,0)])] 
>>> A 
[1, 4, 'x', 8] 
>>> B 
[2, 5, 'y', 9] 
>>> C 
[3, 6, 'z', 0] 

---

在該函數返回一個tup 3個iterables樂，那麼你可以使用itertools.chain從zip扁平化的結果：

from itertools import chain 

A, B, C = (list(chain.from_iterable(i)) 
      for i in zip(*(f(input) 
          for input in iterable))) 

Answer 4

如果A，B和C存在，下面就與返回的值更新他們f：

list(map(lambda x: x[0].extend(x[1]), zip([A,B,C], f())))

list叫，原因map在Python 3