1

我有一个看起来像这样的复杂矩阵：

[[ ['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 项长列表，理论上我想为无限数量的相同大小的组解决这个问题。

4

2 回答 2

2

你可以在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']]
于 2013-09-21T12:02:40.893 回答
1

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

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)
于 2013-09-21T12:08:43.550 回答