Python 商品コードと売上個数の合計

Question

Pythonで4桁の数字からなる商品コードとそれに対応する売上個数が空白で区切られて順に入力されたセットデータを元に商品コードごとの売上個数の合計を求めて画面に出力したい。
ただし商品コードに9999を入力するか100個のセットデータを入力するまで商品コードと売上個数をキーボードより入力する。ここでセットデータの入力は必ず商品コードの昇順で入力しセットデータの個数は1〜100個である、なお商品コードに9999を入力する際には集計の対象とならない任意の売上個数を入力する。
これがどうやればいいのか分かりません。
どうか教えてください。
入力データ例
1001 3
1001 3
1001 1
1002 2
1002 1
1002 5
1002 1
1002 7
1003 8
1003 9
1004 3
1004 5
1004 3
1005 10
1005 2
9999 0

出力例
1001 : 7
1002 : 16
1003 : 17
1004 : 11
1005 : 12

Answer 1

クラスを使って書くとこういう感じです。

class Aggregate(dict):
    def __init__(self, maxcount=100, endcode='9999'):
        self.count = 0
        self.maxcount = maxcount
        self.endcode = endcode

    def entry(self, s):
        code, sales = s.split()
        if code == self.maxcount:
            return False
        if code in self:
            self[code] += int(sales)
        else:
            self[code] = int(sales)
        self.count += 1
        return self.count < self.maxcount

    def __repr__(self):
        return '\n'.join([f'{key} {self[key]}' for key in self])

aggr = Aggregate()
while aggr.entry(input()): pass
print(aggr)

実行結果

>>> aggr = Aggregate()
>>> while aggr.entry(input()): pass
...
1001 3
1001 3
1001 1
1002 2
1002 1
1002 5
1002 1
1002 7
1003 8
1003 9
1004 3
1004 5
1004 3
1005 10
1005 2
9999 0
>>> print(aggr)
1001 7
1002 16
1003 17
1004 11
1005 12

Comments

[退会済みユーザー]

初心者なもので出来ましたらリストまでで作れませんかね。