グラフを作成し保存したいが、エラーが出てしまう…

Question

Python初心者です。グラフを作成し、保存するプログラムを作成中なのですが、グラフの保存ができず、困っています。

なぜグラフの保存ができないのでしょうか？ご親切な方、ご回答お願い致します。

import quandl
import pandas as pd
import  matplotlib.pyplot as plt
import plotly.graph_objects as go
from datetime import date,timedelta

code = 6758

df = pd.read_csv(f"{code}.csv",encoding="shift_jis")


#基準日を設定
day = date(2010,4,16)


#タイムデルタ　－〇日
three_month_ago_delta = timedelta(days=-90)

one_years_ago_delta = timedelta(days=-365) 
one_years_three_month_ago_delta = timedelta(days=-(365+90)) 


#〇日前
three_month_ago = day + three_month_ago_delta
one_years_ago = day + one_years_ago_delta
one_years_three_month_ago = day + one_years_three_month_ago_delta

df_three_month_ago  = df.query(f' "{three_month_ago}" < Date < "{day}" ')
df_one_years_ago  = df.query(f' "{one_years_three_month_ago}" < Date < "{one_years_ago}" ')

print(df_three_month_ago)
print(df_one_years_ago)

fig_df_three_month_ago = go.Figure(
         data= [go.Candlestick(
                   x = df_three_month_ago['Date'], 
                   open=df_three_month_ago['Open'],
                   high=df_three_month_ago['High'],
                   low=df_three_month_ago['Low'],
                   close=df_three_month_ago['Close'])
                 
         ]
)

fig_df_one_years_ago = go.Figure(
         data= [go.Candlestick(
                   x = df_one_years_ago['Date'], 
                   open= df_one_years_ago['Open'],
                   high= df_one_years_ago['High'],
                   low= df_one_years_ago['Low'],
                   close= df_one_years_ago['Close'])
                 
         ]
)

fig_df_three_month_ago.savefig("a.jpg")

print('グラフを保存しました！')

エラーメッセージ

       Date    Open    High     Low   Close      Volume

741 2010-01-18 3035.0 3100.0 3035.0 3090.0 6808400.0
742 2010-01-19 3085.0 3090.0 3045.0 3065.0 5366900.0
743 2010-01-20 3085.0 3115.0 3045.0 3060.0 6318600.0
744 2010-01-21 3060.0 3210.0 3040.0 3185.0 11829800.0
745 2010-01-22 3105.0 3165.0 3095.0 3145.0 11756600.0
.. ... ... ... ... ... ...
798 2010-04-09 3405.0 3440.0 3400.0 3420.0 6803400.0
799 2010-04-12 3460.0 3475.0 3415.0 3415.0 4027500.0
800 2010-04-13 3410.0 3415.0 3350.0 3395.0 5142000.0
801 2010-04-14 3430.0 3455.0 3375.0 3395.0 5854600.0
802 2010-04-15 3445.0 3445.0 3395.0 3410.0 4753500.0

[62 rows x 6 columns]
Date Open High Low Close Volume
498 2009-01-19 2105.0 2135.0 2050.0 2080.0 5549100.0
499 2009-01-20 2025.0 2035.0 1981.0 2025.0 8945900.0
500 2009-01-21 1960.0 2020.0 1950.0 1989.0 11150900.0
501 2009-01-22 2010.0 2020.0 1917.0 1938.0 10610900.0
502 2009-01-23 1850.0 1853.0 1794.0 1802.0 17630100.0
.. ... ... ... ... ... ...
554 2009-04-09 2370.0 2485.0 2370.0 2480.0 10618800.0
555 2009-04-10 2600.0 2610.0 2530.0 2585.0 12527400.0
556 2009-04-13 2560.0 2620.0 2545.0 2555.0 5620700.0
557 2009-04-14 2625.0 2625.0 2520.0 2555.0 6802500.0
558 2009-04-15 2515.0 2530.0 2415.0 2445.0 8865600.0

[61 rows x 6 columns]
Traceback (most recent call last):
File "C:\Users\cjrr6\Desktop\teratail_post.py", line 70, in <module>
fig_df_three_month_ago.savefig("a.jpg")
AttributeError: 'Figure' object has no attribute 'savefig'

Answer 1

【追記】
plotyの場合はfig_df_three_month_ago.write_image("a.jpg")のような書き方になるかと思います。

plotly.io.write_image

---

fig_df_three_month_ago.savefig("a.jpg")ではなく、plt.savefig("a.jpg")ではどうでしょうか？

Comments

[JM235]

夜分遅くにご回答していただき、ありがとうございます。
助言していただいた通りに、plt.savefig("a.jpg")に変更してみました。
エラーメッセージは表示されず、一応画像は保存されましたが、画像にグラフが描写されてませんでした…　真っ白な画像です…

[meg_]

plt.show()をコードの最後にいれた場合、グラフは表示されますか？

[JM235]

ご回答していただき、ありがとうございます。
plt.show()を入れた場合、きちんと表示されます。

[JM235]

fig_df_three_month_ago.write_image("a.jpg")に変更して、プログラムを実行したところ、”ValueError: Image generation requires the psutil package.”と、エラーメッセージが出たので、psutilのインストールをしてみたのですが、またエラーが出てしまい、こちらの解決に手間がかかりそうです…

[JM235]

ただ、おそらくpsutilのインストールができればfig_df_three_month_ago.write_image("a.jpg")で上手くいくと思いますので、とりあえず今回の質問は”解決済”にさせていただきたいと思います。

[JM235]

夜分遅くに何度もご回答いただき、ありがとうございました！