web-dev-qa-db-ja.com

3番目の変数に従って散布図のマーカースタイルを変更する

私は複数列の辞書を扱っています。 2つの列をプロットし、その後、3番目と4番目の列に従ってマーカーの色とスタイルを変更したいと思います。

Pylab散布図のマーカースタイルを変更するのに苦労しています。色に対して機能する私のアプローチは、残念ながらマーカースタイルに対しては機能しません。

x=[1,2,3,4,5,6]
y=[1,3,4,5,6,7]
m=['k','l','l','k','j','l']

for i in xrange(len(m)):
    m[i]=m[i].replace('j','o')
    m[i]=m[i].replace('k','x')
    m[i]=m[i].replace('l','+')

plt.scatter(x,y,marker=m)
plt.show()
13
Krn

問題は、markerは単一の値のみであり、colorパラメーターのようにマーカーのリストではないことです。

マーカー値でグループ化して、同じマーカーを持つxリストとyリストを作成し、それらをプロットすることができます。

xs = [[1, 2, 3], [4, 5, 6]]
ys = [[1, 2, 3], [4, 5, 6]]
m = ['o', 'x']
for i in range(len(xs)):
    plt.scatter(xs[i], ys[i], marker=m[i])
plt.show()

または、すべてのドットをプロットすることもできます(これはお勧めしません)。

x=[1,2,3,4,5,6]
y=[1,3,4,5,6,7]
m=['k','l','l','k','j','l']

mapping = {'j' : 'o', 'k': 'x', 'l': '+'}

for i in range(len(x)):
    plt.scatter(x[i], y[i], marker=mapping[m[i]])
plt.show()
13
Viktor Kerkez

Viktor Kerkezの答えに追加し、Numpyを少し使用すると、次のようなことができます。

x = np.array([1,2,3,4,5,6])
y = np.array([1,3,4,5,6,7])
m = np.array(['o','+','+','o','x','+'])

unique_markers = set(m)  # or yo can use: np.unique(m)

for um in unique_markers:
    mask = m == um 
    # mask is now an array of booleans that van be used for indexing  
    plt.scatter(x[mask], y[mask], marker=um)
10
Jan Kuiken

私にとって最も簡単な解決策は、pandasseabornを使用することでした。

import pandas as pd   # '0.25.3'
import seaborn as sns # '0.9.0'

data = pd.DataFrame(
    dict(x=[1,2,3,4,5,6],
    y=[1,3,4,5,6,7],
    m=['k','l','l','k','j','l'],)
)

sns.scatterplot(data=data, x='x', y='y', style='m')

plot

seabornは自動的にマーカースタイルを選択します。

0
Ian Thompson