2014-10-21 2 views
5

2 개의 외부 벡터의 값에 따라 다른 마커와 다른 색상으로 플롯을 만들고 싶습니다. 내가 시도 여기에서다른 마커와 색상을 가진 matplotlib scatter plot

:

>>> s = [u'+', u'+', u'o'] 
>>> col = ['r','r','g'] 
>>> x = np.array([1,2,3]) 
>>> y = np.array([4,5,6]) 
>>> pl.scatter(x,y,marker=s,c=col) 
Traceback (most recent call last): 
    File "/usr/lib/python3/dist-packages/matplotlib/markers.py", line 233, in set_marker 
    Path(marker) 
    File "/usr/lib/python3/dist-packages/matplotlib/path.py", line 133, in __init__ 
    vertices = np.asarray(vertices, np.float_) 
    File "/usr/lib/python3/dist-packages/numpy/core/numeric.py", line 460, in asarray 
    return array(a, dtype, copy=False, order=order) 
ValueError: could not convert string to float: '+' 

During handling of the above exception, another exception occurred: 

Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
    File "/usr/lib/python3/dist-packages/matplotlib/pyplot.py", line 3087, in scatter 
    linewidths=linewidths, verts=verts, **kwargs) 
    File "/usr/lib/python3/dist-packages/matplotlib/axes.py", line 6298, in scatter 
    marker_obj = mmarkers.MarkerStyle(marker) 
    File "/usr/lib/python3/dist-packages/matplotlib/markers.py", line 162, in __init__ 
    self.set_marker(marker) 
    File "/usr/lib/python3/dist-packages/matplotlib/markers.py", line 236, in set_marker 
    raise ValueError('Unrecognized marker style {}'.format(marker)) 
ValueError: Unrecognized marker style ['+', '+', 'o'] 
>>> 

편집 : 제안 된 솔루션 시뮬레이션 예에서 작동하지만 내 진짜 문제에서 작동하지 않습니다. (나는 왜 그런지 모르겠다).

import matplotlib.pyplot as pl 
import numpy as np 
locY = np.linspace(0, 1, 50) #in the real data locX and locY are coordinate of the points 
locX = np.linspace(0,1,50) 
p = 50 

C = {0: [0, 13, 14, 19, 22, 30], 16: [1, 16, 17], 20: [15, 18, 20, 24], 37: [4, 8, 9, 33, 37, 40, 47], 41: [5, 7, 28, 32, 34, 36, 39, 41, 42, 44, 46], 26: [6, 26, 27, 29, 31, 35], 45: [10, 11, 12, 38, 43, 45, 48, 49], 21: [2, 3, 21, 23, 25]} 
     index = C.keys() 

     def plot_cluster(C, index): 
      color_map = ['b', 'g', 'r', 'c', 'm', 'y', 'k', 'w'] 
      colors = [] 
      m_type = [] 


      for i in range(p): 
       for pos, k in enumerate(index): 
        if i in C[k]: 
         #colors.append(pos) 
         colors.append(color_map[pos]) 
         if i == k: 
          m_type.append(u'+') 
         else: 
          m_type.append(u'o') 


      fig = pl.scatter(locX, locY, c=colors, s=30, cmap='spectral') 
      fig = pl.locator_params(nbins=3) 
      fig = pl.xlim(-0.2, 1.2) 
      fig = pl.ylim(-0.2, 1.2) 
      fig = pl.locator_params(nbins=5, axis='y') 

      return fig 

    pl.show(plot_cluster(C, index)) 

나는 음모에만 2 점을 얻을 제안 된 대답과 함께 음모를 꾸미고 부분은 (내가 왜 모르는) 교체 할 경우

어떻게 해결할 수 : 내 전체 코드는 다음

?

답변

9

이 작동 :

s = [u'+', u'+', u'o'] 
col = ['r','r','g'] 
x = np.array([1,2,3]) 
y = np.array([4,5,6]) 

for _s, c, _x, _y in zip(s, col, x, y): 
    plt.scatter(_x, _y, marker=_s, c=c) 

plt.xlim(0, 4) 
plt.ylim(0, 8) 

plt.show() 

같은 렌더링 :

Plot of above code

업데이트를

당신이 다양한 색깔을 가지고 있고 단일 통화를 할 수 있습니다 보인다 분산 기능 : . 다중 색상 기능은 API에서 확인되었지만 마커 kwarg에 대한 반복 가능을 지정할 수 있음을 읽지 않습니다. 제거하면 코드가 작동합니다. marker=s

+0

이 예제는 시뮬레이션 된 예제이지만 실제 코드에서는 작동하지 않습니다. 질문을 업데이트했습니다 – Donbeo

+0

해당 코드를 실행하려고하면 "AttributeError : '모듈'객체에 'show'속성이 없습니다 (즉, 마지막 행에 문제가 있음). 다시 확인할 수 있습니까 –

+0

그것은 나를 위해 작동합니다. 들여 쓰기에주의하고 import matplotlib.pyplot을 pl로 사용하고 numpy를 처음에 np로 가져 오기 – Donbeo

관련 문제