3

我想将一个字符串列表连接成按列表中的值分组的新字符串。这是我的意思的一个例子：

输入

key = ['1','2','2','3']
data = ['a','b','c','d']

结果

newkey = ['1','2','3']
newdata = ['a','b c','d']

我了解如何加入文本。但我不知道如何正确迭代列表的值以聚合相同键值共有的字符串。

任何帮助或建议表示赞赏。谢谢。

4

5 回答 5

10 
from collections import defaultdict

d = defaultdict(list)

for k, v in zip(key, data):
    d[k].append(v)

print [(k, ' '.join(v)) for k, v in d.items()]

输出：

[('1', 'a'), ('3', 'd'), ('2', 'b c')]

以及如何获取新列表：

newkey, newvalue = d.keys(), [' '.join(v) for v in d.values()]

并保存订单：

newkey, newvalue = zip(*[(k, ' '.join(d.pop(k))) for k in key if k in d])
于 2013-01-11T14:27:05.820 回答
2

使用itertools.groupby()功能组合元素；zip 将允许您将两个输入列表分组为两个输出列表：

import itertools
import operator

newkey, newdata = [], []
for key, items in itertools.groupby(zip(key, data), key=operator.itemgetter(0)):
    # key is the grouped key, items an iterable of key, data pairs
    newkey.append(key)
    newdata.append(' '.join(d for k, d in items))

你可以把它变成一个更zip()神奇的列表理解：

from itertools import groupby
from operator import itemgetter

newkey, newdata = zip(*[(k, ' '.join(d for _, d in it)) for k, it in groupby(zip(key, data), key=itemgetter(0))])

请注意，这确实需要对输入进行排序；groupby仅根据相同的连续键对元素进行分组。另一方面，它确实保留了最初的排序顺序。

于 2013-01-11T14:27:01.047 回答
0

只是为了多样化，这里有一个没有任何外部库和字典的解决方案：

def group_vals(keys, vals):
    new_keys= sorted(set(keys))
    zipped_keys = zip(keys, keys[1:]+[''])
    zipped_vals = zip(vals, vals[1:]+[''])
    new_vals = []
    for i, (key1, key2) in enumerate(zipped_keys):
        if key1 == key2:
            new_vals.append(' '.join(zipped_vals[i]))
        else:
            new_vals.append(zipped_vals[i][0])
    return new_keys, new_vals

group_vals([1,2,2,3], ['a','b','c','d'])
# --> ([1, 2, 3], ['a', 'b c', 'd'])

但我知道它很丑陋，可能不如其他解决方案那么高效。仅用于演示目的。:)

于 2013-01-11T15:14:44.277 回答
0

你可以itertools.groupby()使用zip(key,data)：

In [128]: from itertools import *

In [129]: from operator import *

In [133]: lis=[(k," ".join(x[1] for x in g)) for k,g in groupby(zip(key,data),key=itemgetter(0))]

In [134]: newkey,newdata=zip(*lis)

In [135]: newkey
Out[135]: ('1', '2', '3')

In [136]: newdata
Out[136]: ('a', 'b c', 'd')
于 2013-01-11T14:26:27.457 回答
0

如果您不想导入集合，您可以随时使用常规字典。

key = ['1','2','2','3']
data = ['a','b','c','d']
newkeydata = {}

for k,d in zip(key,data):
    newkeydata[k] = newkeydata.get(k, []).append(d)
于 2013-01-11T14:31:37.120 回答