列表列表或多個列表可以元素組合的所有方式是什么 - 對每個列表中具有相同索引的項目進行操作？

Question

我有一個列表列表，並希望對元素進行操作，將具有相同索引的項目組合在一起。

a = [[1,2,3],
     [4,5,6],
     [7,8,9]]

# or
b = [['1', '2', '3'],
     ['4', '5', '6'],
     ['7', '8', '9']]

對a[0][i] 、 a[1][i] 、 a[2][i] 。

new = [f(a[0][0],a[1][0]a[2][0]),
       f(a[0][1],a[1][1],a[2][1]),
       f(a[0][2],a[1][2],a[2][1])]

其中f是任何接受適當參數的函數。

---

嘗試為多個類似問題制定規范。 隨意添加我忽略的答案。

Answer 1

首先使用 zip 來組合項目。

多個列表。

>>> list(zip(['1', '2', '3'], ['4', '5', '6'], ['7', '8', '9']))
[('1', '4', '7'), ('2', '5', '8'), ('3', '6', '9')]

使用zip(*thelist) - rows,columns -> columns,rows - 轉置列表列表 -

a = [[1,2,3],
     [4,5,6],
     [7,8,9]]

>>> list(zip(*a))
[(1, 4, 7),
 (2, 5, 8),
 (3, 6, 9)]

b = [['1', '2', '3'],
     ['4', '5', '6'],
     ['7', '8', '9']]

>>> list(zip(*b))
[('1', '4', '7'),
 ('2', '5', '8'),
 ('3', '6', '9')]

完成后，您可以迭代和組合。 都有一個相似的模式：調用傳遞組合值的函數/方法。

求和數

>>> [sum(thing) for thing in zip(*a)]
[12, 15, 18]
>>> list(map(sum, zip(*a)))
[12, 15, 18]

添加/連接/連接字符串

>>> [''.join(thing) for thing in zip(*b)]
['147', '258', '369']
>>> list(map(''.join, zip(*b)))
['147', '258', '369']

將字符串列表與數字列表相結合。

>>> x = ['a', 'b', 'c']
>>> y = [1,2,3]
>>> z = ['g', 'e', 'h']
>>> [''.join(str(item) for item in thing) for thing in zip(x,y,z)]
['a1g', 'b2e', 'c3h']
>>> [''.join(map(str,thing)) for thing in zip(x,y,z)]
['a1g', 'b2e', 'c3h']

將函數應用於元素列表。

>>> def f(args):
...     a,b,c,*_ = args
...     return (a+b)**c

>>> [f(thing) for thing in zip(*a)]
[78125, 5764801, 387420489]