pythonでの配列の処理

Question

事前に準備された以下の食材のリストがあります。

pre_shopping_list =[['じゃがいも','１個'],['玉ねぎ','２個'],['豚肉','200g'],['人参','１本'],['キャベツ','１個'],['玉ねぎ','２個'],['じゃがいも','２個']]

複数回登録されている食材があるので、その様な項目については、食材の種類ごとに個数（分量）をまとめて最終的に以下の様に買い物をする際に見やすいリストを作成したいと考えています。（分量の足し算などはしません）

shopping_list = [['じゃがいも', '１個', '２個'], ['玉ねぎ', '２個', '２個'], ['豚肉', '200g'], ['人参', '１本'], ['キャベツ', '１個']]

現在のコードとその結果得られるshopping_listは以下の通りです。

shopping_list = [['','']]

for i in range(len(pre_shopping_list)):
  for j in range(len(shopping_list)):
      if shopping_list[j][0]==pre_shopping_list[i][0]:
          shopping_list[j].append(pre_shopping_list[i][1])   
  shopping_list.append(pre_shopping_list[i])      
shopping_list.pop(0)

結果

shopping_list = [['じゃがいも', '１個', '２個'], ['玉ねぎ', '２個', '２個'], ['豚肉', '200g'], ['人参', '１本'], ['キャベツ', '１個'], ['玉ねぎ', '２個'], ['じゃがいも', '２個']]

for文の使い方が悪いのか、その場合どの様に修正すれば良いのか、もしくはそもそもfor分では不可能なのか分からず困っています。ご教授お願い致します。

Answer 1

配列->配列で重複項目を判断するのは大変なので、一度dictを中継すると楽に処理できるようになりそうです。

pre_shopping_list =[['じゃがいも','１個'],['玉ねぎ','２個'],['豚肉','200g'],['人参','１本'],['キャベツ','１個'],['玉ねぎ','２個'],['じゃがいも','２個']]

shopping_dict = {}

for item in pre_shopping_list:
    shopping_dict.setdefault(item[0], [])
    shopping_dict[item[0]].append(item[1])

shopping_list = [[k] + v for k, v in shopping_dict.items()]
print(shopping_list)

一度、shopping_dictで品目をキーに個数をまとめておき、最後にlistにすることで実装が比較的スッキリします。

Comments

[sandalwalk]

回答ありがとうございます。
shopping_list = [[k] + v for k, v in shopping_dict.items()]
でのkは作成した辞書のキー、vは値だと理解できるのですが、この書き方で、自動的にshopping_dict.items()の２つの変数を処理してくれるのですね。これが内包表現というものでしょうか？

Answer 2

ソートしてからgroupbyが楽です。

import itertools
import operator

car = operator.itemgetter(0)
shopping_list = [
    [key, *(e for _, e in it)]
    for key, it in itertools.groupby(
        sorted(pre_shopping_list, key=car), key=car
    )
]
print(shopping_list)

辞書を使うなら、defaultdictが便利です。

import collections

shopping_dict = collections.defaultdict(list)
for name, amount in pre_shopping_list:
    shopping_dict[name].append(amount)
    
shopping_list = [
    [key, *value] for key, value in shopping_dict.items()
]
print(shopping_list)

Answer 3

shopping_list.append(pre_shopping_list[i])

がかならず実行されるので、こういう結果になります。
原型を残したければフラグを使って内側のfor文内で追加した場合には、この追加をしなければ良いのですが、綺麗ではありません。

辞書を使うのが良いでしょう。

>>> pre_shopping_list =[['じゃがいも','１個'],['玉ねぎ','２個'],['豚肉','200g'],['人参','１本'],['キャベツ','１個'],['玉ねぎ','２個'],['じゃがいも','２個']]
>>> shopping_dict = {}
>>> for item, num in pre_shopping_list:
...     if item in shopping_dict:
...         shopping_dict[item].append(num)
...     else:
...         shopping_dict[item] = [num]
...
>>> shopping_list = [[item] + shopping_dict[item] for item in shopping_dict]
>>> print(shopping_list)
[['じゃがいも', '１個', '２個'], ['玉ねぎ', '２個', '２個'], ['豚肉', '200g'], ['人参', '１本'], ['キャベツ', '１個']]