X軸の範囲によってプロットの色を変えたい

Question

お世話になっております。

###前提・実現したいこと
２次元散布図において、x軸の範囲によってプロットの色を変えて同一グラフにプロットしたい。

###発生している問題・エラーメッセージ
下記のエラーコードが出てしまい、プログラムが実行できない。

plt.scatter(X[idx] , Y[idx] , **kwargs)
TypeError: 'float' object has no attribute '__getitem__'

###該当のソースコード

        x2 = (round(((pMax1[0] + width1/2)-385)*0.327,2))
        y2 = (round(((pMax1[1] + height1/2)-290)*0.327,2))
        x3 = (round(((pMax2[0] + width1/2)-385)*0.327,2))
        y3 = (round(((pMax2[1] + height1/2)-290)*0.327,2)) 
        l = np.linalg.norm(b-a)
        l = (round(l,2))
        X = (round((50/(l-50))*((x2 + x3)/2),2))
        Y = (round((50/(l-50))*((y2 + y3)/2),2))
        X1 = [17.80,17.80,18.20,18.00]
        X2 = [-23.00,-23.30,-23.30,-23.30]
        X3 = [0.00,0.00,0.00,0.00,0.00,0.00]
        Y1 = [13.50,10.20,7.10,4.00]
        Y2 = [14.50,11.20,8.10,5.00]
        Y3 = [0.00,-7.50,-15.25,-21.10,-27.00,-32.75]
        
        plt.scatter(X1, Y1, s=800,facecolors="none", marker="o", linewidths="0.1",edgecolors="r")
        plt.scatter(X2, Y2, s=800,facecolors="none", marker="o", linewidths="0.1",edgecolors="b")
        plt.scatter(X3, Y3, s=800,facecolors="none", marker="o", linewidths="0.1",edgecolors="g")

        def PENEFAX1(X, Y, rangeX, **kwargs):
            idx = np.where((X >= rangeX[0]) & (X < rangeX[1]))
            plt.scatter(X[idx] , Y[idx] , **kwargs)
        PENEFAX1(X, Y, rangeX=(-30, -10), c='r',marker='o')
        PENEFAX1(X, Y, rangeX=(-10, 10), c='b', marker='o')
        PENEFAX1(X, Y, rangeX=(10, 30), c='g', marker='o')

色々なHPや質問サイトを参考に関数を組むことが得策だということで組んでみましたが上手くいきませんでした。

###試したこと

x2,y2,x3,y3,l,X,YはOpenCVのテンプレートマッチングで動画上の座標位置を計算していて、動画の再生と併せて座標位置の計算結果(X,Y)をグラフ上に散布図としてプロットされていました。

X1,X2.X3,Y1,Y2,Y3(こちらも色分けするために無理やり分けています)とX,Yを1つのグラフにプロットしたいです。

###補足情報(言語/FW/ツール等のバージョンなど)
windows10
python2.7
matplotlib
spyder

Answer 1

エラーの原因は、X、Yの値としてfloat(浮動小数点数)を渡しているのに、

X = (round((50/(l-50))*((x2 + x3)/2),2))
PENEFAX1(X, Y, rangeX=(-30, -10), c='r',marker='o')

メソッド内で配列として参照しようとしている箇所があるためです。

def PENEFAX1(X, Y, rangeX, **kwargs):

idx = np.where((X >= rangeX[0]) & (X < rangeX[1]))
plt.scatter(`X[idx]` , `Y[idx]` , **kwargs)

idxの結果によりプロットする、しないを分けたいということであれば、
if文による条件分岐等が必要となるのではないでしょうか。

Comments

[Penefax]

ご回答ありがとうございます。「idxの結果によりプロットする、しないを分けたいということであれば、
if文による条件分岐等が必要となるのではないでしょうか。」という点に関しては特にこだわりはありません。正直、「idx = np.where((X >= rangeX[0]) & (X < rangeX[1]))」の意味がいまいち理解できていません。