7

有没有更好的方法来迭代我的字典数据而不使用 3 嵌套for loops，就像我目前在给定下面的数据所做的那样？顺便说一句，我使用的是 python 2.6。

data = {'08132012': 
           {
            'id01': [{'code': '02343','status': 'P'},{'code': '03343','status': 'F'}],
            'id02': [{'code': '18141','status': 'F'},{'code': '07777','status': 'F'}]
           }
        }

这是3个for循环当前代码：

  for date in data:
      for id in data[date]:
          for trans in data[date][id]:
              print "Date: %s" % date
              print "Processing id: %s" % id
              print trans['code']
              print trans['status']

              //query to database

已编辑：有效数据值

4

3 回答 3

8

鉴于数据的嵌套性质，我认为没有任何方法可以避免某处的一些嵌套循环。

但是，您可以通过为数据编写扁平化生成器来避免嵌套大部分程序逻辑，如下所示：

def flatten(data):
    for date in data:
        for id in data[date]:
            for trans in data[date][id]:
                yield (date, id, trans)

def process(data):
    for date, id, trans in flatten(data):
        # do stuff with date, id, trans
于 2012-08-25T06:03:36.417 回答
4

在不了解数据来源的更多细节的情况下，我可能会考虑使用namedtuple是否可以更改data存储方式的结构。它将使事情变得更干净，更具可读性。

这是一个例子：

>>> from collections import namedtuple
>>> Record = namedtuple('Record', ['date', 'id', 'code', 'status'])
>>> records = []
>>> records.append(Record('08132012', 'id01', '02343','P'))
>>> records.append(Record('08132012', 'id01', '03343','F'))
>>> records.append(Record('08132012', 'id02', '18131','F'))
>>> records.append(Record('08132012', 'id02', '07777','F'))
>>> for record in records:
...     print "Date: %s" %record.date
...     print "Processing id: %s" %record.id
...     print record.code
...     print record.status
... 
Date: 08132012
Processing id: id01
02343
P
Date: 08132012
Processing id: id01
03343
F
Date: 08132012
Processing id: id02
18131
F
Date: 08132012
Processing id: id02
07777
F

更多乐趣：

获取status'F' 所在的记录列表：

>>> Fs = [record for record in records if record.status == 'F']
>>> Fs
[Record(date='08132012', id='id01', code='03343', status='F'),
Record(date='08132012', id='id02', code='18131', status='F'),
Record(date='08132012', id='id02', code='07777', status='F')]

排序方式code：

>>> records.append(Record('08122012', 'id03', '00001', 'P'))
>>> records.sort(key=lambda x:x.code)
>>> records
[Record(date='08122012', id='id03', code='00001', status='P'),
 Record(date='08132012', id='id01', code='02343', status='P'),
 Record(date='08132012', id='id01', code='03343', status='F'),
 Record(date='08132012', id='id02', code='07777', status='F'),
 Record(date='08132012', id='id02', code='18131', status='F')]
于 2012-08-25T06:28:13.440 回答
4

生成器相当简单：

>>> flat = ((d, id, t) for d in data for id in data[d] for t in data[d][id])
>>> for date, id, trans in flat:
...     print date, id, trans
...
08132012 id02 {'status': 'F', 'code': '18141'}
08132012 id02 {'status': 'F', 'code': '07777'}
08132012 id01 {'status': 'P', 'code': '02343'}
08132012 id01 {'status': 'F', 'code': '03343'}
>>>
于 2012-08-25T09:20:46.300 回答