6

我有一个这样的清单

[u'201003', u'200403', u'200803', u'200503', u'201303',
 u'200903', u'200603', u'201203', u'200303', u'200703', u'201103']

让我们将此列表称为“years_list”

当我按年分组时，

group_by_yrs_list = groupby(years_list, key = lambda year_month: year_month[:-2]) 
for k,v in group_by_yrs_list:
  print k, list(v)

我得到了想要的输出：

2010 [u'201003']
2004 [u'200403']
2008 [u'200803']
2005 [u'200503']
2013 [u'201303']
2009 [u'200903']
2006 [u'200603']
2012 [u'201203']
2003 [u'200303']
2007 [u'200703']
2011 [u'201103']

然后，我像这样稍微改变了我的实现，

  group_by_yrs_list = dict(groupby(years_list, key = lambda year_month: year_month[:-2]))
  for k,v in group_by_yrs_list.items():
    print k, list(v)

我刚刚添加了一个dict，但是输出不同，

2003 []
2006 []
2007 []
2004 []
2005 []
2008 []
2009 []
2011 [u'201103']
2010 []
2013 []
2012 []

我找不到原因。请帮我找出字典实际上在做什么。

(Python 2.7)

4

4 回答 4

9

groupby产生成对的 (key, iterator -of-group)。如果您正在迭代第二对，则第一对的迭代器组已经被消耗，因此您得到空列表。

试试下面的代码：

group_by_yrs_list = {year:list(grp) for year, grp in groupby(years_list, key=lambda year_month: year_month[:-2])}
for k, v in group_by_yrs_list.items():
    print k, v
于 2013-10-01T06:19:38.877 回答
6

这里的问题是groupby依次产生每个键和一个子迭代器：

>>> for k, v in groupby(years_list, key = lambda year_month: year_month[:-2]):
...    print k, v
2010 <itertools._grouper object at 0x801c68950>
2004 <itertools._grouper object at 0x801bb3a90>
2008 <itertools._grouper object at 0x801c68950>
2005 <itertools._grouper object at 0x801bb3a90>
2013 <itertools._grouper object at 0x801c68950>
2009 <itertools._grouper object at 0x801bb3a90>
2006 <itertools._grouper object at 0x801c68950>
2012 <itertools._grouper object at 0x801bb3a90>
2003 <itertools._grouper object at 0x801c68950>
2007 <itertools._grouper object at 0x801bb3a90>
2011 <itertools._grouper object at 0x801c68950>

您需要将每个<itertools._grouper object ...>列表转换为实际列表，然后再将其存储起来，因为下一次迭代会groupby重置迭代器。如果你不这样做，只剩下一个有用的迭代器，所以当你打印字典的内容时，你会得到一个非空列表（它用完迭代器）。再次打印它，您将获得全空列表。

关键是在迭代器仍然很好的时候列出它们（我看到其他几个人比我更喜欢示例代码，我更喜欢falsetru 的变体）。

于 2013-10-01T06:21:40.887 回答
2

尝试groupby从toolz

$ pip install toolz
$ ipython

In [1]: from toolz import groupby

In [2]: years_list = [u'201003', u'200403', u'200803', u'200503', u'201303',
   ...:  u'200903', u'200603', u'201203', u'200303', u'200703', u'201103']

In [3]: get_year = lambda year_month: year_month[:-2]

In [4]: groupby(get_year, years_list)
Out[4]: 
{u'2003': [u'200303'],
 u'2004': [u'200403'],
 u'2005': [u'200503'],
 u'2006': [u'200603'],
 u'2007': [u'200703'],
 u'2008': [u'200803'],
 u'2009': [u'200903'],
 u'2010': [u'201003'],
 u'2011': [u'201103'],
 u'2012': [u'201203'],
 u'2013': [u'201303']}
于 2013-10-28T01:06:58.437 回答
1

根据this answer，您可以这样做将其转换为dict：

group_by_yrs_list = dict((k,list(v)) for k,v in groupby(years_list, key=lambda x: x[:4]))

这是因为 的输出groupby是一个itertools.groupby对象，它是一种生成器，显然不能直接用作dict构造函数的参数。

于 2013-10-01T06:20:07.483 回答