python - 大神们，求个算法-PHP中文网问答

讲师中心微信公众号

首页

文章

后端开发 web前端数据库开发工具 php框架常见问题人工智能 Java 系统教程电脑教程硬件教程手机教程软件教程游戏教程自媒体新闻

专题

后端开发 web前端数据库开发工具 php框架人工智能 Java 系统教程电脑教程硬件教程手机教程软件教程游戏教程新闻

AI工具

AI 聊天问答 Agent智能体 AI 文本写作 AI 绘画作图 AI 设计工具 AI 视频创作 AI 音频制作 AI 办公学习 AI 编程开发 Prompt指令

学习

大前端后端开发数据库移动端运维开发计算机基础

编程手册

大前端后端开发数据库移动端运维开发计算机基础

下载

js特效网站源码工具下载类库下载网站素材学习资源插件扩展手机游戏

最近更新

python - 大神们，求个算法

黄舟 2017-04-18 09:17:38

[Python讨论组]

287

现有如下数据结构:

data = {
  "date": {
    "buckets": [
      {
        "key": "2015-01-01",
        "shop": {
          "buckets": [
            {
              "key": "A",
              "sum_qty": {
                "value": 5
              },
              "sum_amt": {
                "value": 10
              }
            }
          ]
        }
      },
      {
        "key": "2015-06-01",
        "shop": {
          "buckets": [
            {
              "key": "B",
              "sum_qty": {
                "value": 20
              },
              "sum_amt": {
                "value": 100
              }
            }
          ]
        }
      }
    ]
  }
}

最终结果：

lst = [
    {'date': '2015-01-01', 'shop': 'A', 'sum_qty': 5, 'sum_amt': 10},
    {'date': '2015-06-01', 'shop': 'B', 'sum_qty': 20, 'sum_amt': 100}
]
print lst

黄舟

人生最曼妙的风景，竟是内心的淡定与从容！

全部回复(3)

高洛峰2017-04-18 09:19:38 3楼

寫了一個 class, 結果應該跟你要的一樣:

from collections import abc

class CoolJSON:

    def __init__(self, key, mapping):
        """
        key is the main key of this CoolJSON
        mapping is the total data
        """
        self.key = key
        self.mapping = dict(mapping)


    def collect_bucket_item(self, item):
        """ used to handle single bucket item"""
        dic = {}
        lst = [{}]

        for key, value in item.items():
            if key=='key':
                dic[self.key] = value
            elif isinstance(value, abc.MutableMapping):
                if 'buckets' in value:      
                    lst = CoolJSON(key, value).collect()
                elif 'value' in value:
                    dic[key] = value['value']

        for item in lst:
            item.update(dic)

        return lst


    def collect(self):
        """used to collect results from all bucket items"""
        results = []
        for item in self.mapping['buckets']:
            results.extend(self.collect_bucket_item(item))
        return results

測試:

# 測資使用 prolife 在評論下方新給的測資

lst = CoolJSON('date', data['date']).collect()

for item in lst:
    print(item)

結果:

{'sum_amt': 3651755, 'sum_qty': 36927, 'date': '2015-02-01', 'shop': 'A'}
{'sum_amt': 436019, 'sum_qty': 12115, 'date': '2015-02-01', 'shop': 'B'}
{'sum_amt': 1310549, 'sum_qty': 8896, 'date': '2015-02-01', 'shop': 'C'}
{'sum_amt': 18620841, 'sum_qty': 212909, 'date': '2015-03-01', 'shop': 'A'}
{'sum_amt': 5105368, 'sum_qty': 40109, 'date': '2015-03-01', 'shop': 'B'}
{'sum_amt': 938102, 'sum_qty': 28156, 'date': '2015-03-01', 'shop': 'C'}

我知道這可能不是你要的答案, 不過我不是很清楚你遇到的資料的狀況, 所以我先拋出一版, 再來看怎麼調整到可以處理你的需求, 以下是一個把這份資料想得很單純的無腦寫法:

lst = []

for item in data['date']['buckets']:
    dic = {}
    dic['date'] = item['key']
    dic['shop'] = item['shop']['buckets'][0]['key']
    dic['sum_qty'] = item['shop']['buckets'][0]['sum_qty']['value']
    dic['sum_amt'] = item['shop']['buckets'][0]['sum_amt']['value']
    lst.append(dic)

print(lst)

我回答過的問題: Python-QA