JSON形式のデータを条件付きで集計したい

前提・実現したいこと

下記のJSONデータから'event_id'と'event_name'があるものだけ、
'event_name'ごとに集計し、'event_id'と'event_name'とあわせて表示させたいです。
数の集計はできたのですが、'event_id'と'event_name'を同じ{}内に表示させるのにうまくいかず悩んでいます。

###json_data

[
  {
    "id": "001",
    "no": "001",
    "timestamp": 1617185234,
    "event_id": 2,
    "event_name": "b"
  },
  {
    "id": "001",
    "no": "1010",
    "timestamp": 1618296823,
    "event_id": 3,
    "event_name": "c"
  },
  {
    "id": "001",
    "no": "102",
    "timestamp": 1617185290
  },
  {
    "id": "002",
    "no": "102",
    "timestamp": 1617186107
  }
....

出力結果

[
  "{"a_count": 10}",
  "{"b_count": 20}",
  "{"c_count": 30}",
  "{"d_count": 40}"
]

該当のソースコード

def counter():
    a_count = Counter(v for d in json_data for v, v in d.items() if str(v).startswith('a'))
    b_count = Counter(v for d in json_data for v, v in d.items() if str(v).startswith('b'))
    c_count = Counter(v for d in json_data for v, v in d.items() if str(v).startswith('c'))
    d_count = Counter(v for d in json_data for v, v in d.items() if str(v).startswith('d'))
    showcount = json.dumps(a_count),json.dumps(b_count),json.dumps(c_count),json.dumps(d_count)
return show_count

実現したいこと

[
    { "event_id": 1, "event_name": "a" ,"a_count": 10},
    { "event_id": 2, "event_name": "b", "b_count": 20} ,
    { "event_id": 3, "event_name": "c", "c_count": 30},
    { "event_id": 4, "event_name": "d", "d_count": 40}
]

文字列として追加させようとしたのですが、

"event_id": 1, "event_name": "a" ,{"a_count": 10}

となってしまいやめました。。

8524ba23

2021/10/11 01:15

event_nameが同じでevent_idが異なる複数の要素がある場合は、どのような結果としたいですか？たとえば「{ "event_id": [1,2], "event_name": "a", "a_count" : 2 }」のような感じなど。

mu250003

2021/10/11 01:25

質問ありがとうございます！はい、event_nameが同じでevent_idが異なる場合は "event_id": [1,2]と表示させたいです。

行動規範の内容に同意します

回答1件

ベストアンサー

元のデータからキーがevent_name, 値がevent_idのリストとなる辞書を作成して、その辞書から結果リストを作成します。
なお、同一event_idは１つにまとめたいようなので、そのようにしています。

Python
1import json
2
3s = """
4[ {"event_id":1, "event_name":"a"},
5  {"event_id":2, "event_name":"a"},
6  {"event_id":2, "event_name":"a"},
7  {"event_id":3, "event_name":"b"},
8  {"hoge":123} ]"""
9json_data = json.loads(s)
10
11dct = {} # キー=event_name, 値=event_idのリスト
12for data in json_data:
13    if 'event_id' in data and 'event_name' in data:
14        key = data['event_name']
15        if key not in dct:
16            dct[key] = []
17        dct[key].append(data['event_id'])
18
19ret = []
20for key in sorted(dct.keys()):
21    ids = dct[key]
22    d = {}
23    ids = sorted(list(set(ids))) # 同じidはまとめる。（先の辞書作成時にやっておいたほうがよいが）
24    d['event_id'] = ids
25    d['event_name'] = key
26    d[f'{key}_count'] = len(ids)
27    ret.append(d)
28
29print(ret)
30# [{'event_id': [1, 2], 'event_name': 'a', 'a_count': 2}, {'event_id': [3], 'event_name': 'b', 'b_count': 1}]

投稿2021/10/11 01:31

編集2021/10/11 09:10

8524ba23

総合スコア38341

mu250003

2021/10/11 08:23

ご回答ありがとうございます！試してみたところ、 { "event_id": [ 4, 4, 4, 4, 4, 4, 4, 4 ], "event_name": "a", "a_count": 8 となってしまいました。 d['event_id'] = ids　を　d['event_id'] = set(ids) にしたりしましたが、だめでした。。 event_idをdistinctするにはどうしたらよいでしょうか？