Pandasのplotを使った際に描画されないデータがある。

Question

現在下記のデータが格納されているCSVファイルを読み込んで描画するプログラムを作成しております。

Time[us],data1,data2
0,NaN,1
246.3,0,NaN
778.8,0,NaN
1778.8,0,NaN
2778.8,NaN,1
3778.8,0,NaN
4778.8,NaN,0
5278.8,0,NaN
6718.8,NaN,1
7718.8,NaN,0

このデータを下記のコードで描画すると一部データが描画されません。

import pandas as pd
import matplotlib.pyplot as plt
import matplotlib
import seaborn

df = pd.read_csv("test.csv")
df.plot(x = "Time[us]")
plt.show()

図を見るとdata1の5378.8 μsの値と、data2の0μsの値が描画されていないように思われます。
これらの値を含めて描画したいのですが、どのようにしたらいいでしょうか。

不足している情報などございましたらご教示お願いいたします。

Answer 1

NaNが挟まった部分はプロットされません。そして前後をNaNに挟まれているポイントでは線は引けません。

マーカーを出せばとりあえずマーカーは見えるようになるので、もしかしたらそれが目的のものかもしれません。

import io
import pandas as pd
import matplotlib.pyplot as plt

txt = """
Time[us],data1,data2
0,NaN,1
246.3,0,NaN
778.8,0,NaN
1778.8,0,NaN
2778.8,NaN,1
3778.8,0,NaN
4778.8,NaN,0
5278.8,0,NaN
6718.8,NaN,1
7718.8,NaN,0
"""

df = pd.read_csv(io.StringIO(txt))
df.plot(x="Time[us]", marker=".")
plt.savefig("result.png")

result.png

間を繋げたい場合、DataFrame.plot一発だといけない気がします。いろいろやり方は思いつきます。df[["Time[us]", "data1"]]とdf[["Time[us]", "data2"]]をそれぞれ作り、NaNがある行を落として、同じAxesオブジェクトに対してプロットするというロジックで書くのが一つの手です。このやり方でやるとこんな感じです。

import io
import pandas as pd
import matplotlib.pyplot as plt

txt = """
Time[us],data1,data2
0,NaN,1
246.3,0,NaN
778.8,0,NaN
1778.8,0,NaN
2778.8,NaN,1
3778.8,0,NaN
4778.8,NaN,0
5278.8,0,NaN
6718.8,NaN,1
7718.8,NaN,0
"""

df = pd.read_csv(io.StringIO(txt))

ax = plt.subplot()
for col in ["data1", "data2"]:
    tmp = df[["Time[us]", col]]
    tmp.dropna(axis=0, how="any").plot(x="Time[us]", ax=ax)

plt.savefig("result.png")

result.png

Comments

[kina_programing]

丁重に答えていただきありがとうございます。
無事やりたいことができました。