pythonで連続したラベルのみ折れ線でつなぎたい

Question

データファイルからラベル番号，x座標，y座標の三つを取得したときに，
連続するラベル番号の時のみ，x，y座標からプロットを作成しプロット間に線を結びたいです．

プログラミング言語の扱いを十分に理解できていないため失礼な質問であるかもしれませんが，どうかよろしくお願いします．

list = {"label": [1,2,3,4,5,6,7,10,11,12,13,18,19,20], 
        "x": [1,2,3,4,5,6,7,10,11,12,13,18,19,20],
        "y": [1,2,3,4,5,6,7,10,11,12,13,18,19,20]
        }

Answer 1

以下のような感じで連続する部分毎にグラフ描画すると、不連続になります。

import pandas as pd
import matplotlib.pyplot as plt

d = {"label": [1,2,3,4,5,6,7,10,11,12,13,18,19,20], 
        "x": [1,2,3,4,5,6,7,10,11,12,13,18,19,20],
        "y": [1,2,3,4,5,6,7,10,11,12,13,18,19,20]
        }
df = pd.DataFrame(d)

# 連続する部分（グループ）に番号を振る
df['grp'] = df['label'] != df['label'].shift(1) + 1
df.loc[0,'grp'] = False
df['grp'] = df['grp'].cumsum()

# グループ毎にグラフを描画
grp = sorted(set(df['grp']))
for g in grp:
    dfg = df[df['grp'] == g]
    plt.plot(dfg['x'], dfg['y']) # 色を統一したい場合はc='blue'などを指定する

plt.show()

Comments

[doraemon_pha]

ご教授いただきありがとうございます．
実際に扱おうとしているデータは直線的なものでなく複数の渦を巻くようなプロットになるため，一つ一つのグループのプロットの遷移の仕方から解析を行いたいと考えていました．
そのため，グループごとにデータを区切る方法を知ることができ大変感謝しております．ありがとうございます．

すみません，pandasでデータフレームによりグループを管理されていると思うのですが，特定のグループを呼び起こしてその１グループのみをプロットすることも可能になるのでしょうか．

知識も経験もなくすべて頼ってしまうような形となってしまい申し訳ありません．

[can110]

はい。できます。
回答コードではすべてのグループをひとつのグラフに描画していますが、特定のグループだけをグラフに描画、あるいはそれぞれのグループを異なるグラフに描画(subplot)することもできます。

[doraemon_pha]

ご回答ありがとうございます．
早速コードして動作が確認できました．
ありがとうございます．

Answer 2

求めていらっしゃるのはこういうグラフでしょうか。

import matplotlib.pyplot as plt

data = {"label": [1,2,3,4,5,6,7,10,11,12,13,18,19,20], 
        "x": [1,2,3,4,5,6,7,10,11,12,13,18,19,20],
        "y": [1,2,3,4,5,6,7,10,11,12,13,18,19,20]
        }

label = data['label']
x =  data['x']
y =  data['y']
x_plot = []
y_plot = []
for i in range(min(label), max(label) + 1):
    if i in label:
        x_plot.append(x[label.index(i)])
        y_plot.append(y[label.index(i)])
    else:
        x_plot.append(None)
        y_plot.append(None)

print(x_plot)
print(y_plot)
plt.plot(x_plot, y_plot)
plt.show()

実行結果

なお、listという名前は組込みクラスの名前ですので、変数としては使わない方が安全です。

Comments

[doraemon_pha]

ご教授いただきありがとうございます．
Noneを与えると不連続になることを始めて知りました．

listという名前は組込みクラスの名前ですので、変数としては使わない方が安全です。
＞当方変数の名前についても理解が追い付いていない状況でした．ご指摘ありがとうございます．