pythonで描いたグラフに横線(plotの定義)を入れたい、そして複数に書かれたデータをcsvに一括で落としたい

Question

pythonで描いたグラフに横線を入れたい、そして複数に書かれたデータをすべて同じcsvの同じシートに一括で落としたいのですが、上手くいきません

質問が2つあります。

①pythonで描いたグラフで、イールドスプレッドは左軸、QQQは右軸にして、左軸が「0」の値で横線を入れたい

②数に書かれたデータをすべて同じcsvの同じシートに一括で落としたい

〔自分が実行したこと〕
①に関してはコードを書くと添付のようにすべて右軸で表示され、「plot.axhline(y = 0, color = 'red', ls = 'dotted')」をいれると、エラーが表示されます

②に関しては、色々考え、探したのですが、コードが見つかりませんでした

コード
import numpy as np
import pandas as pd
import datetime as dt
import pandas_datareader.data as web
import yfinance as yf

s = dt.date(2000, 1, 1)
e = dt.date(2022, 3, 18)



df_2 = web.DataReader("DGS2", "fred", s, e)
df_3 = web.DataReader("DGS3", "fred", s, e)
df_5 = web.DataReader("DGS5", "fred", s, e)
df_7 = web.DataReader("DGS7", "fred", s, e)
df_10 = web.DataReader("DGS10", "fred", s, e)
df_20 = web.DataReader("DGS20", "fred", s, e)
df_30 = web.DataReader("DGS30", "fred", s, e)

codelist = ["QQQ"]

df_QQQ = yf.download(codelist, s, e)["Adj Close"]

df = pd.concat([df_2,df_3, df_5, df_7,df_10, df_20, df_30,df_QQQ], axis=1)

#イールドスプレッドの定義付け

df['spread5-2'] = df['DGS5'] - df['DGS2']
df['spread10-2'] = df['DGS10'] - df['DGS2']
df['spread10-3'] = df['DGS10'] - df['DGS3']
df['spread10-7'] = df['DGS10'] - df['DGS7']
df['spread30-5'] = df['DGS30'] - df['DGS5']
df['spread30-20'] = df['DGS30'] - df['DGS20']


#チャート、イールドスプレッド

df['spread5-2'].plot(figsize=(15,10),fontsize=15)
df['spread10-2'].plot(figsize=(15,10),fontsize=15)
df['spread10-3'].plot(figsize=(15,10),fontsize=15)
df['spread10-7'].plot(figsize=(15,10),fontsize=15)
df['spread30-5'].plot(figsize=(15,10),fontsize=15)
df['spread30-20'].plot(figsize=(15,10),fontsize=15)
df_QQQ.plot(figsize=(15,10),fontsize=15)

plot.axhline(y = 0, color = 'red', ls = 'dotted')

#CSVに落とす
df['spread5-2'].to_csv('spread5-2.csv')
df['spread10-2'].to_csv('spread10-2.csv')
df['spread10-3'].to_csv('spread10-3.csv')
df['spread10-7'].to_csv('spread10-7.csv')
df['spread30-20'].to_csv('spread30-20.csv')
df_QQQ.to_csv('df_QQQ.csv')

Comments

[meg_]

> ①に関してはコードを書くと添付のようにすべて右軸で表示され、「plot.axhline(y = 0, color = 'red', ls = 'dotted')」をいれると、エラーが表示されます
エラーとはどんなエラーでしょうか？

[mufufu]

NameError: name 'plot' is not defined

上記のエラーが表示されます

[meg_]

> NameError: name 'plot' is not defined
> 上記のエラーが表示されます
jupyterでコード実行されているようなので「plot」を定義したセルの実行漏れではありませんか？

※質問は編集できますので情報は質問に追記した方が回答が付きやすくなるかと思います。

[mufufu]

はい、jupyterでかいました。

この場合は、どのようにplotの定義をするのでしょうか。

ネットで検索したコードをもとにかいたのですが、人によっては「plot」と「plt」の意味がおなじなのでしょうか。

一般的に「plt」と「plot」は違う意味なのでしょうか。

[melian]

エラーに関しては以下で表示される様になります。
ax = df_QQQ.plot(figsize=(15,10),fontsize=15)
ax.axhline(y = 0, color = 'red', ls = 'dotted')

[meg_]

> この場合は、どのようにplotの定義をするのでしょうか。
引用元が分からないので分かりません。

> ネットで検索したコードをもとにかいたのですが、人によっては「plot」と「plt」の意味がおなじなのでしょうか。
> 一般的に「plt」と「plot」は違う意味なのでしょうか。
おそらくmatplotlib.pyplotをimportする際によく使われる「import matplotlib.pyplot as plt」のことを指しているかと思いますが、pltが使用されることが多いかと思います。（別に他の名前でも良いです）

[mufufu]

QQQチャートを除いたチャートでは下記のようにして、横線が表示されました。

ax = df['spread5-2'].plot(figsize=(15,10),fontsize=15)
ax.axhline(y = 0, color = 'red', ls = 'dotted')

QQQのデータを右軸としてチャートを描くにはどうしたいいですか。

Answer 1

①pythonで描いたグラフで、イールドスプレッドは左軸、QQQは右軸にして、左軸が「0」の値で横線を入れたい

twinx() を使います。
※ 左右の区別がしづらかったので line color を設定しています

#チャート、イールドスプレッド

# df['spread5-2'].plot(figsize=(15,10),fontsize=15)
# df['spread10-2'].plot(figsize=(15,10),fontsize=15)
# df['spread10-3'].plot(figsize=(15,10),fontsize=15)
# df['spread10-7'].plot(figsize=(15,10),fontsize=15)
# df['spread30-5'].plot(figsize=(15,10),fontsize=15)
# df['spread30-20'].plot(figsize=(15,10),fontsize=15)

ax1 = df['spread5-2'].plot(figsize=(15,10),fontsize=15,color='blue')
ax1.axhline(y = 0, color = 'red', ls = 'dotted')
ax2 = ax1.twinx()
df_QQQ.plot(figsize=(15,10),fontsize=15,color='green',ax=ax2)

②数に書かれたデータをすべて同じcsvの同じシートに一括で落としたい

df に全てのデータ(イールドスプレッド + QQQ)が含まれていますので、そのまま CSV ファイルとして保存します。

#CSVに落とす
df.to_csv('df_all.csv')

Comments

[mufufu]

全部できました。

csvは↓↓で全部落ちるんですね。
「df.to_csv('df_all.csv')」

凡例表示も加えようとして、下記のコードを加えたのですが、

ax1 = df['spread5-2'].plot(figsize=(15,10),fontsize=15,color='blue')
ax1.axhline(y = 0, color = 'red', ls = 'dotted')
ax2 = ax1.twinx()
df_QQQ.plot(figsize=(15,10),fontsize=15,color='green',ax=ax2)

plot.legend(fontsize=12,loc='upper left')

「NameError: name 'plot' is not defined」
と、表示されてしまいました。
これはどのように対処したらいいですか。

[melian]

それぞれのプロットに label= を追加します。そして、ax1 と ax2 の legend をマージします。

ax1 = df['spread5-2'].plot(figsize=(15,10),fontsize=15,color='blue',label='spread5-2')
ax1.axhline(y = 0, color = 'red', ls = 'dotted')
ax2 = ax1.twinx()
df_QQQ.plot(figsize=(15,10),fontsize=15,color='green',ax=ax2,label='QQQ')

#凡例(legend)
lines1, labels1 = ax1.get_legend_handles_labels()
lines2, labels2 = ax2.get_legend_handles_labels()
ax1.legend(lines1 + lines2, labels1 + labels2, loc='upper left')

[mufufu]

ありがとうございました。