Question

我有一个复杂的矩阵，如下所示：

[[ ['x', '1', '2', '3', '4'],
   ['y', '5', '6', '7', '8']],

 [ ['x', 'a', 'b', 'c', 'd'],
   ['y', 'e', 'f', 'g', 'h']  ] ]

我想把它变成这个：

['x', '1a', '2b', '3c', '4d'],
['y', '5e', '6f', '7g', '8h']

我正在破坏我的头脑，但没有达到目的。此外，即使我只有两组嵌套的5项长列表，理论上我想为无数个相同大小的组解决这个问题。

Answer 1

您可以在此处使用dict：

>>> from operator import add
>>> lis = [[ ['x', '1', '2', '3', '4'],
   ['y', '5', '6', '7', '8']],
 [ ['x', 'a', 'b', 'c', 'd'],
   ['y', 'e', 'f', 'g', 'h']  ] ]
>>> dic = {}
for item in lis:
    for x in item:
        k, v = x[0], x[1:]
        if k in dic:
            dic[k] = map(add, dic[k], v)
        else:
            dic[k] = v
...             
>>> dic
{'y': ['5e', '6f', '7g', '8h'], 'x': ['1a', '2b', '3c', '4d']}
#list of lists
>>> [[k] + v for k, v in dic.iteritems()]
[['y', '5e', '6f', '7g', '8h'], ['x', '1a', '2b', '3c', '4d']]

使用zip，reduce和列表理解的另一种解决方案：

>>> from operator import add
>>> def func(x, y):
...     return map(add, x, y[1:])

>>> [[item[0][0]] + reduce(func, item[1:], item[0][1:])  for item in zip(*lis)]
[['x', '1a', '2b', '3c', '4d'], ['y', '5e', '6f', '7g', '8h']]

Answer 2

这是一个“有趣”的解决方案。由于您没有提供有关阵列结构的任何信息，因此我假设最简单的变体：

import numpy

a = numpy.array([[
        ['x', '1', '2', '3', '4'],
        ['y', '5', '6', '7', '8']],
    [
        ['x', 'a', 'b', 'c', 'd'],
        ['y', 'e', 'f', 'g', 'h']]],
    dtype=numpy.object)

res = a[0].copy()
for chunk in a[1:]:
    res[:,1:] += chunk[:,1:]

print(res)

使用python合并复杂矩阵

2 个答案: