python openpyxl 分類毎の合計を出したい

Question

Pythonのopenpyxlを使って種類毎の合計を出したいと思ってます。

下記のようなExcelの表があるとして、種類ごとの個数の合計を出したいです。
種類 金額
りんご 1
りんご 1
みかん 1
みかん 1
ぶどう 1
ぶどう 1
柿　　 1
柿 　　　1
いちご 1
いちご 1
メロン 1
メロン 1
すいか 1
すいか 1
レモン 1
レモン 1
個数　りんご：2　　みかん：2・・・

自分で考えて書いたコードが下記となります。
wb = openpyxl.load_workbook(ファイルの絶対パス)
ws = wb[Sheet1]

SUM = 0

List=["りんご”,"みかん,"ぶどう”,"柿”,"いちご”,"メロン”,"すいか”]

for i in List:
if ws.cell(row=i,column=1)=＝List
for j in range(2:ws.max_row):
SUM=sum+j

すみません、行き詰ってしまいましたのでどなたか教えていただければと思います。　

Comments

[melian]

Pandas の read_excel を使ってデータフレーム化してみてはどうですか。

[退会済みユーザー]

pandasではうまくいきました！for i, group in df.groupby("種類"):
openpyxlの知識を増やしたくて・・・

[melian]

なるほど、了解です。

[takasima20]

Excel からデータを取り出す処理と集計する処理を分けて考えてみては?

Answer 1

Excel の Formula(計算式)を使う方法もあります(Excel ファイルに保存する必要がありますけれども)。

import openpyxl

wb = openpyxl.load_workbook('test.xlsx')
ws = wb['Sheet1']

# sum by groups
max_row = ws.max_row
row_sum = max_row + 2
col = 66
ws[f'A{row_sum}'] = '個数'
for i in range(2, ws.max_row):
    if ws[f'A{i}'].value == ws[f'A{i-1}'].value:
        ws[f'{chr(col)}{row_sum}'] = f'=CONCATENATE(A{i},":",SUMIF(A:A,A{i},B:B))'
        col += 1

# save
wb.save('test_sums.xlsx')

test_sums.xlsx

追記

import openpyxl
from collections import defaultdict
from pprint import pprint

wb = openpyxl.load_workbook('test.xlsx')
ws = wb['Sheet1']

item_counts = defaultdict(int)
for i, j in ws[f'A2:B{ws.max_row}']:
    item_counts[i.value] += j.value
item_counts = dict(item_counts)

pprint(item_counts, width=20, sort_dicts=False)

#
{'りんご': 2,
 'みかん': 2,
 'ぶどう': 2,
 '柿': 2,
 'いちご': 2,
 'メロン': 2,
 'すいか': 2,
 'レモン': 2}

Answer 2

openpyxlでやるなら以下です。

import openpyxl
wb = openpyxl.load_workbook('fruit.xlsx')
ws=wb['Sheet1']

result = {}
for i in range(2, ws.max_row):
    v = ws.cell(i, 1).value
    if not v in result:
        result[v] = 1
    else:
        result[v] += 1

print(result)

pandasを使うなら以下です。

df = pd.read_excel('fruit.xlsx')
result = df.groupby('種類').sum()
print(result)

Comments

[退会済みユーザー]

ありがとうございます！ただ、上の通りコードを書いたところ、TypeError: string indices must be integers　のエラーが出てしまいました・・調査した結果、eval()を使うようなのですがどこに使えばよいかわからず、大変申し訳ないですが教えていただけますか。