python - 在 Python 中如何选择列表中包含重复项的行

Question

我有一个列表，其中包含如下列表:

rowdata = [['100004100', 'ABC1234AS', '45.96'],
           ['100004101', 'ABC1234AS', '104.95'],
           ['100004103', '453SDFAS', '24.52']]

我正在尝试弄清楚如果第二个值匹配，如何将第一个值和第三个值作为一个组返回。

第一个值是订单号，第二个值是批号，第三个是金额。我的信用卡处理商将随机订单分批处理。我试图找到批号匹配的行并返回这些行的订单号和金额。

我在想，

for item in rowdata:
    if item[1] is a duplicate:
        print item[0], item[1]

Answer 1

Python 有一些非常有用的内置工具，但有时如果您不知道要查找什么，则很难找到它们。这里groupby和 itemgetter很有帮助。

from itertools import groupby
from operator import itemgetter

rowdata = [['100004100', 'ABC1234AS', '45.96'],
           ['100004101', 'ABC1234AS', '104.95'],
           ['100004103', '453SDFAS', '24.52']]

for key, group in groupby(rowdata, itemgetter(1)):
    print key, list(group)

给你:

ABC1234AS [['100004100', 'ABC1234AS', '45.96'], ['100004101','ABC1234AS', '104.95']]

453SDFAS [['100004103', '453SDFAS', '24.52']]