如何使用python将两个列表组合成一个列表

Question

我有两个列表，其中包含多个元组。我正在寻找一种将两个列表合并为一个列表的方法。

list 1 = [('mike','Company A','developer'),('steve','Company B','developer'),('tom','Company B','tester'),('jerry','Company A','tester')]

list 2 = [('mike','Company A','C++'),('steve','Company B','Python'),('tom','Company B','Automation'),('mike','Company A','Manual')]

Expected output:

list 3 = [('mike','Company A','developer','C++'),('steve','Company B','developer,'Python'),('tom','Company B','tester','Automation'),('mike','Company A','tester','Manual')

Answer 1

合并这两个列表很容易：

list1 + list2
# [('mike', 'Company A', 'developer'),
#  ('steve', 'Company B', 'developer'),
#  ('tom', 'Company B', 'tester'),
#  ('jerry', 'Company A', 'tester'),
#  ('mike', 'Company A', 'C++'),
#  ('steve', 'Company B', 'Python'),
#  ('tom', 'Company B', 'Automation'),
#  ('mike', 'Company A', 'Manual')]

然后对它们进行排序。

>>> sorted(list1 + list2, key=lambda x: (x[0], x[1]))
# [('jerry', 'Company A', 'tester'),
#  ('mike', 'Company A', 'developer'),
#  ('mike', 'Company A', 'C++'),
#  ('mike', 'Company A', 'Manual'),
#  ('steve', 'Company B', 'developer'),
#  ('steve', 'Company B', 'Python'),
#  ('tom', 'Company B', 'tester'),
#  ('tom', 'Company B', 'Automation')]

使用 itertools.groupby 按前两个元素对它们进行分组。

>>> groupby(sorted(list1 + list2, key=lambda x: (x[0], x[1])), key=lambda x: (x[0], x[1]))
<itertools.groupby object at 0x7f77deb2f818>

使用列表理解从每个组中的每个值中提取第三个元素。

>>> [(k[0], k[1], [x[2] for x in v]) for k, v in groupby(sorted(list1 + list2, key=lambda x: (x[0], x[1])), key=lambda x: (x[0], x[1]))]
# [('jerry', 'Company A', ['tester']),
#  ('mike', 'Company A', ['developer', 'C++', 'Manual']),
#  ('steve', 'Company B', ['developer', 'Python']),
#  ('tom', 'Company B', ['tester', 'Automation'])]

从那里开始，只需将这些列表扩展到每个元组中即可。

Answer 2

list3 = list1+list2

连接两个列表