値の大きさで色の濃さを指定 plot

Question

値ごとに色の濃さを変えたい

前回の質問に対する回答者のおかげで、値によって色の濃さを変えてplotすることはできました。

今回行いたいことは、メンバーシップ値(01)の値の大きさで濃さを指定したいです。
例えば、メンバーシップ値が00.3の時は、薄く、0.3~0.6は少し濃くなど

clusteringはクラスタリングの結果で、クラスタ番号を指します。

該当のソースコード

def visualizeClustering(data, clustering, eps, minPoints, maxPoints,memberships):
    
    c_labels = list(set(clustering))
    m_labels = list(set(membership))
    
    # Eliminates multiple values
    # Color for each cluster
    colors = plt.cm.Spectral(np.linspace(0, 1, len(c_labels)))
    alphas = np.linspace(0, 1,len(m_labels))
   
    cs = [(l,colors[i]) for i,l in enumerate(c_labels)]
    ms = [(l,alphas[i]) for i,l in enumerate(m_labels)]
    
    for c,m in itertools.product(cs, ms):
    # ２つのラベルで抽出
        if c[0] == -1:
            for i in range(0,len(c_labels)):
                c[1][i]=0
               
        mask = (clustering == c[0]) & (membership == m[0])
        xy = data[mask]    

        plt.plot(xy[:, 0], xy[:, 1], 'o', markerfacecolor=c[1], markeredgecolor='k', alpha=m[1], markersize=7)

plt.show()

試したこと

alphas=list((0.3,0.5,0.7,1))のように濃さを作成し、
if文で各点のmembershipごとにalpahsを指定しようとしました。

教えていただいたコードを崩さないようにしたら、分からなくなってしまいました。

Answer 1

例えば、メンバーシップ値が00.3の時は、薄く、0.30.6は少し濃くなど

そのような透明度を返す関数を作り、その関数で透明度を決めればよいです。

import matplotlib.pyplot as plt
import numpy as np
import itertools

np.random.seed(110)
N = 20
data = np.random.randint(1,100, size=(N,2))
clustering = np.random.randint(1,3,N)
membership = np.random.rand(N)

#例えば、メンバーシップ値が0~0.3の時は、薄く、0.3~0.6は少し濃くなど
def get_alpha(v):
    if v <= 0.3:
        return 0.3
    elif v <= 0.6:
        return 0.5
    return 1

c_labels = list(set(clustering))
m_labels = list(set(membership))

colors = plt.cm.Spectral(np.linspace(0, 1, len(c_labels)))

# ラベルと値（色, 透明度）の組み合わせ
cs = [(l,colors[i]) for i,l in enumerate(c_labels)]
ms = [(l,get_alpha(l)) for l in m_labels] # 関数で透明度を決める

for c,m in itertools.product(cs, ms):
    # ２つのラベルで抽出
    mask = (clustering == c[0]) & (membership == m[0])
    xy = data[mask]
    plt.plot(xy[:, 0], xy[:, 1], 'o', markerfacecolor=c[1], markeredgecolor='k', alpha=m[1], markersize=7)

plt.show()

Comments

[wabi_4]

確かにできました。
前回の質問に引き続き、誠にありがとうございます。
とても助かりました。