matplotlibのグラフの書き方

Question

実現したいこと

pythonを使用しています。

a = [[0,0],[1,1],[1,2],[0,3],[0,4],[0,5],[0,6],[1,7]]

上記のデータを
横軸にa[i][1]、縦軸にa[i][0]の値をとって
matplotlibを用いてグラフを書きたいのですが、どうにもうまくいきません。

どうすればよいでしょうか。

試したこと

a = [[0,0],[1,1],[1,2],[0,3],[0,4],[0,5],[0,6],[1,7]]

a0 = []
a1 = []
for i in range(len(a)):
    a0.append(a[i][0])
    a1.append(a[i][1])
plt.plot(a1,a0)

このグラフを1になった時だけ上がるような図にしたいです。
下記が理想図です。

Answer 1

pyplot.plot() のパラメータに drawstyle='steps-mid' (またはsteps-preorsteps-post)を設定すると良いかと思います。

import matplotlib.pyplot as plt

a = [[0,0],[1,1],[1,2],[0,3],[0,4],[0,5],[0,6],[1,7]]

y,x = zip(*a)
plt.plot(x,y,drawstyle='steps-mid')
plt.show()

Comments

[TeRa_130]

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

Answer 2

magichanさんの回答がグラフを描画するテクニックとして適切だと思います。
自分はそのオプションを知らなかったので、元データを自前で工夫するという迂遠な方法をかいてしまいました

---

plot関数は「与えられていないxの値」について「その値に近い隣の点のf(x)の値を線形補完」します。質問者さんのコード中のデータでは

x = 0 => f(x) = 0
x = 1 => f(x) = 1

というような情報しかないのでplotは 0 <= x <= 1 の区間におけるxに対するf(x)の値は0から1へ緩やかに変化するものと想定して描画します。

要するに元データ[0, 0], [1, 1]では

xが1の近傍で急激に0->1に変化する

という情報が一切含まれてないのが理想どうりにならない理由です。
理想図に近いものをplotで描きたいなら、「x=1の近傍でf(x)が0->1へ急激に変化する」ことを表すようなデータを与えてください。例えば

x	f(x)
0.00	0.0
0.99	0.0
1.00	1.0
...	...

このようなデータを与えればx=1の近傍でf(x)が0から1へ急激に変化する様子を表現できます。

Comments

[TeRa_130]

わかりやすい解説ありがとうございます。