3

我需要遍历一个已排序的数据集，将所有结果按该排序属性分组为具有该属性相同值的块。然后我对该块结果运行一些操作。

抱歉，这有点令人困惑，示例可能是描述我正在做的事情的更好方式：

我有一个结构类似这样的数据集，除了“数据”字符串实际上是对象并包含大量其他数据。

[ [1, "data1"], [1, "data2"], [2, "moredata"], [2, "stuff"], 
  [2, "things"], [2, "foo"], [3, "bar"], [4, "baz"] ]

我想要发生的是将这些数据分组为 4 个不同的函数调用：

process_data(1, ["data1", "data2"])
process_data(2, ["moredata", "stuff", "things", "foo"])
process_data(3, ["bar"])
process_data(4, ["baz"])

我最终得到的是一个看起来像这样的构造：

last_id = None
grouped_data = []

for row in dataset:
    id = row[0]
    data = row[1]

    if last_id != id:
         # we're starting a new group, process the last group
         processs_data(last_id, grouped_data)
         grouped_data = []
    last_id = id
    grouped_data.append(data)

if grouped_data:
    # we're done the loop and we still have a last group of data to process
    # if there was no data in the dataset, grouped_data will still be empty
    # so we won't accidentally process any empty data.
    process_data(last_id, grouped_data)

它有效，但看起来很笨拙。特别是需要使用 last_id 变量以及循环后对 process_data 的第二次调用来跟踪所有内容。我只想知道是否有人可以为更优雅/更聪明的解决方案提供任何建议。

我选择的语言是 Python，但一般的解决方案很好。

4

3 回答 3

5

itertools.groupby 正是你想要的：

>>> data = [ [1, "data1"], [1, "data2"], [2, "moredata"], [2, "stuff"],
...   [2, "things"], [2, "foo"], [3, "bar"], [4, "baz"] ]
>>>
>>> from itertools import groupby
>>> from operator import itemgetter
>>>
>>> def process_data(key, keydata):
...     print key, ':', keydata
...
>>> for key,keydata in groupby(data, key=itemgetter(0)):
...   process_data(key, [d[1] for d in keydata])
...
1 : ['data1', 'data2']
2 : ['moredata', 'stuff', 'things', 'foo']
3 : ['bar']
4 : ['baz']

传递 groupby 一个排序列表，以及一个关于在列表中的每个项目中分组依据的关键函数。你得到一个(key,itemgenerator)对的生成器，如图所示被传递给我编造的 process_data 方法。

于 2012-08-06T06:21:20.463 回答
3

看看itertools.groupby。请注意，这要求您的列表已经根据组键（您的示例数据是，所以我想没关系）进行排序。

于 2012-08-06T06:16:57.527 回答
3

您可以使用 MutliDict，例如来自brownie或werkzeug包。

from brownie.datastructures import MultiDict
data = [ [1, "data1"], [1, "data2"], [2, "moredata"], [2, "stuff"], 
         [2, "things"], [2, "foo"], [3, "bar"], [4, "baz"] ]
for key, keydata in MultiDict(data).iterlists():
    process_data(key, keydata)
于 2012-08-06T06:24:01.350 回答