Python 辞書型で重複する値があった場合の結合について

Question

お世話になっております。

上のようなリストからitecCodeの値が同じであれば下にあるように[jan][DateId]にある値を結合したいのですが何か良い方法は無いでしょうか？

調べつつ色々試してみたのですが、うまく下記のように結合できる方法が実現出来なかったため何か良い方法があれば教えて頂けませんでしょうか。

宜しくお願い致します。

list = [
    {'itemCode': 'aaa',    
  'jan': '999999',      
  'DateId': '13'},
    {'itemCode': 'bbb',
  'jan': '888888',
  'DateId': '41'},
    {'itemCode': 'aaa',
  'jan': '999991',
  'DateId': '41'},
    {'itemCode': 'ccc',
  'jan': '777777',
  'DateId': '38'}
]

実現したいもの

list = [
    {'itemCode': 'aaa',    
  'jan': [999999,999991],      
  'DateId': [13,41]},
    {'itemCode': 'bbb',
  'jan': '888888',
  'DateId': '41'},
    {'itemCode': 'ccc',
  'jan': '777777',
  'DateId': '38'}
]

Comments

[LouiS0616]

実際に試したコードを追記して下さい。
どんな方法をどのように試してどのように失敗したのか読み取れません。

Answer 1

DataFrameにする方法は既出なので、リストでチマチマやる方法。

新しいリストにappendしつつ、すでに同じitemCodeが新しいリストに存在するときには、janとDateIdの値をappendする(janとDateIdの値がリストでない場合はリスト化も行なう)。

lst = [
    {'itemCode': 'aaa', 'jan': '999999', 'DateId': '13'},
    {'itemCode': 'bbb', 'jan': '888888', 'DateId': '41'},
    {'itemCode': 'aaa', 'jan': '999991', 'DateId': '41'},
    {'itemCode': 'ccc', 'jan': '777777', 'DateId': '38'}
]

new_lst = []
for dic in lst:
    for dic2 in new_lst:
        if dic['itemCode'] == dic2['itemCode']:
            if type(dic2['jan']) is not list:
                dic2['jan'] = [dic2['jan']]
            dic2['jan'].append(dic['jan'])

            if type(dic2['DateId']) is not list:
                dic2['DateId'] = [dic2['DateId']]
            dic2['DateId'].append(dic['DateId'])
            break
    else:
        new_lst.append(dic)

print(new_lst)

[{'itemCode': 'aaa', 'jan': ['999999', '999991'], 'DateId': ['13', '41']}, {'itemCode': 'bbb', 'jan': '888888', 'DateId': '41'}, {'itemCode': 'ccc', 'jan': '777777', 'DateId': '38'}]

Comments

[AbcdefgFirst]

ご回答ありがとうございます！

頂いたコードを試したところ希望通りの出力になりました。
最後のelseの部分が分からないのですが、同じインデントにifが無い場合も使用出来るのでしょうか？

elseを外してインデントを1つ浅くしたところ、同じitemCodeの商品があった場合にjan、DateIdをリストにはまとめられたのですが、その商品分もnew_lstにも値が追加されてしまいました。

[Daregada]

最後のelseは、他の言語ではあまり見かけない「forに対するelse」です。
途中でbreakしないでforループを最後まで処理したときにだけelse以下が実行されます。

[AbcdefgFirst]

ありがとうございます。
このような書き方は知らなかったので勉強になりました(_ _)

Answer 2

itertools.groupby なども検討しましたが、pandas の DataFrame にして、希望する形式に変更してから、再度 dict に戻すのが簡単そうです。

1. dict を DataFrame にする
2. groupby("itemCode").aggregate(list) で itemCode が同じ要素はまとめる
3. (もし必要であれば)要素が1つしかないものはリストを外す。['888888'] のように要素が1つのリストになってもよいのであればこの処理は不要。
4. to_dict("records") で dict に戻す。

import pandas as pd
from pprint import pprint

lst = [
    {"itemCode": "aaa", "jan": "999999", "DateId": "13"},
    {"itemCode": "bbb", "jan": "888888", "DateId": "41"},
    {"itemCode": "aaa", "jan": "999991", "DateId": "41"},
    {"itemCode": "ccc", "jan": "777777", "DateId": "38"},
]


df = pd.DataFrame(lst)

# itemCode が同じ辞書をまとめる。
ret = df.groupby("itemCode").aggregate(list).reset_index()

# 要素が1つしかないものはリストを外す。
ret = ret.applymap(lambda x: x[0] if isinstance(x, list) and len(x) == 1 else x)

# dict にする。
ret = ret.to_dict("records")

pprint(ret)
# [{'DateId': ['13', '41'], 'itemCode': 'aaa', 'jan': ['999999', '999991']},
#  {'DateId': '41', 'itemCode': 'bbb', 'jan': '888888'},
#  {'DateId': '38', 'itemCode': 'ccc', 'jan': '777777'}]

Comments

[AbcdefgFirst]

ご回答ありがとうございます！
Pandasは使った事が無いのですが頂いたコードを試したところ動作致しました。
今後の参考にさせて頂きたいと思います。ありがとうございます。