蟒蛇繁殖明智兩個列表元素 - 其中之一是嵌套

Question

我有兩個列表：

C = [3, 2, 1] 
D = [[0, 1, 2, 3], [0, 1, 2], [0, 1]] 

我想實現這樣的結果：

E = [[0, 3, 6, 9], [0, 2, 4], [0, 1]] 

只使用列表comprehensions.Is可能？我被困在：

E = zip(C, D) 
[i * E[0][0] for i in E[0][1]] which gives: 

[0, 3, 6, 9] 

，但我不能修改申請第二個列表

Answer 1

你幾乎沒有其他元素：它是一個嵌套列表理解。巴掌另一個列表，[嗒嗒[於J]爲J]

C = [3, 2, 1] 
D = [[0, 1, 2, 3], [0, 1, 2], [0, 1]] 

E = zip(C, D) 
print [[i * E[j][0] for i in E[j][1]] for j in range(len(E))] 

輸出：

[[0, 3, 6, 9], [0, 2, 4], [0, 1]] 

Answer 2

是的，這是可能的：

>>> [[c * d_i for d_i in d] for c, d in zip(C, D)] 
[[0, 3, 6, 9], [0, 2, 4], [0, 1]] 

你只需要兩個循環在這裏，第一個遍歷zip(C, D)，第二個迭代遍歷每個列表D

Answer 3

這個wou LD是直線前進，如果你使用numpy數組：

import numpy as np 
np.array(C) * np.array(map(np.array, D)) 

# array([array([0, 3, 6, 9]), array([0, 2, 4]), array([0, 1])], dtype=object) 

Answer 4

我也將與itertools.startmap解決這個問題：

>>> from itertools import starmap 
>>> 
>>> def multiply(x, lst): 
     return [x*i for i in lst] 

>>> for item in it.starmap(f, zip(C,D)): 
     print(item) 


[0, 3, 6, 9] 
[0, 2, 4] 
[0, 1] 

OR：

>>> l = [] 
>>> for item in it.starmap(lambda x,y: [x*i for i in y], zip(C,D)): 
     l.append(item) 
>>> l 
[[0, 3, 6, 9], [0, 2, 4], [0, 1]]