アップロードされたExcelファイルを基にグラフを作成したい

Question

前提・実現したいこと

Python(Flask,Matplotlib)で、アップロードされたExcelファイルを基にグラフを作成したい

①WebページからExcelテンプレートファイルをダウンロード


②値を入力したExcelファイルを選択し、送信


③uploadsディレクトリに格納される

④アップロードされたExcelファイルを元にグラフを作成

①~③は問題なく動作しています。
④は完成イメージです。

エラーメッセージ・該当のソースコード

 if request.method == 'POST':
 　　file = request.files['file']
 　　UPLOAD_FOLDER = r'C:\xxx\xxx\xxx\TestProject\uploads'
 　　app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
       　if file:
           filename = secure_filename(file.filename)
           file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
           wb = xlrd.open_workbook(r"C:\xxx\xxx\xxx\TestProject\uploads\test.xlsx")
           sheets = wb.sheets()
           month = np.array(sheets['Sheet1'].cell(row=i, column=0).value for i in range(1, 13))
           sale = np.array(sheets['Sheet1'].cell(row=i, column=1).value for i in range(1, 13))
           cost = np.array(sheets['Sheet1'].cell(row=i, column=2).value for i in range(1, 13))
           benefit = sale - cost
           #プロットの処理が続く

エラー例①

 sheets = wb.sheets()
 month = np.array([sheets['Sheet1'].cell(row=i, column=0).value for i in range(1, 13)])
 sale = np.array([sheets['Sheet1'].cell(row=i, column=1).value for i in range(1, 13)])
 cost = np.array([sheets['Sheet1'].cell(row=i, column=2).value for i in range(1, 13)])
 benefit = sale - cost
 plt.plot(month, sale)
 plt.plot(month, cost)
 plt.plot(month, benefit)

   # month行でエラー
→　TypeError: list indices must be integers or slices, not str

エラー例②

 sheets = wb.sheets()
 month = np.array(sheets['Sheet1'].cell(row=i, column=0).value for i in range(1, 13))
 sale = np.array(sheets['Sheet1'].cell(row=i, column=1).value for i in range(1, 13))
 cost = np.array(sheets['Sheet1'].cell(row=i, column=2).value for i in range(1, 13))
 benefit = sale - cost
 plt.plot(month, sale)
 plt.plot(month, cost)
 plt.plot(month, benefit)

　 #benefit行でエラー
→　TypeError: unsupported operand type(s) for -: 'generator' and 'generator'

試したこと

下記のコードは、正しく動作し冒頭にある④のグラフが作成されます

w = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])
x = np.array([10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10, 10])
y = np.array([7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7, 7])
z = x - y

plt.plot(w, x, label='sale')
plt.plot(w, y, label='cost')
plt.plot(w, z, label='benefit')

環境

開発環境：PyCharm
サーバ　：localhost
ブラウザ：Chrome

Comments

[tiitoi]

month, sale, cost, benefit の変数を print して値がどうなっているか見てみてはどうでしょうか？

[Majespy]

month初期化の次行にprint("abc")print(month)を追加しましたが、何も表示されませんでした。アップロードされたExcelファイルの読み込みが間違っているのでしょうか。

[tiitoi]

そうですね。matplotlib の関数の使い方は問題なさそうなので、与えるデータに原因がありそうです。print(month) でなにも表示されないのはおかしいかと思います。month 以外にも sale, cost, benefit もちゃんと値が取れているのか確認したほうがよいですね。

[Majespy]

sale, cost, benefitも同様の結果になりました。ファイルをアップロードさせる処理が今回初の試みであり、アップロードされたファイルの読み込みはエラーがでないように書き直しを繰り返したのでその点がうまくいってないのでしょうか。(エラーメッセージ・該当のソースコードにファイル読み込みのソースを追加しました)

[tiitoi]

wb = xlrd.open_workbook() で開いた Excel ファイルを直接 Excel で開いたら、ちゃんと中身が入っているか確認してみてはどうでしょうか？入っていれば、次は xlrd で値を取り出すところに問題があるということになるので、そのように問題を切り分けていきましょう。

[Majespy]

局所化して考えるというアドバイスにより問題が解決されました。ありがとうございました。

Answer 1

①rangeの範囲指定訂正
②xlrd.open_workbookではなくopenpyxlのload_workbookを使用

上記の方法で、問題が解消されました。

wb = load_workbook(r".\uploads\test.xlsx")
sheet = wb['Sheet1']
month = np.array([sheet.cell(row=i, column=1).value for i in range(2, 14)])