2016-12-15 2 views
0

Я хочу использовать несколько цветов в маркере, выполненном с помощью matplotlib. Выполнение двух цветов было не так сложно, следуя this example, а также с дополнительной информацией от this documentation. Однако мне было интересно, можно ли сделать маркер с более чем 2 цветами. Я в ситуации, когда я хочу, чтобы один маркер фактически получал 3 разных цвета (точка на карте относится к трем различным наблюдениям).Множество цветных заливок в марке matplotlib

+0

http://matplotlib.org/examples/api/scatter_piecharts.html – tom

+0

@tom делает это также работать с нормальными участками (plt.plot (..))? –

+0

@tom или другие маркеры –

ответ

1

Вы можете сделать это, следуя Matplotlib примера, показанного здесь:

matplotlib.org/examples/api/scatter_piecharts.html

Ниже я изменил пример немного использовать ax.plot вместо ax.scatter.

В основном это означает, что весь ваш маркер должен иметь тот же размер, и вместо использования s kwarg для scatter, вы используете ms (или markersize) kwarg для plot.

Кроме того, вместо facecolor вам необходимо определить markerfacecolor.

Помимо этих изменений, все остальное остается таким же, как в исходном примере.

""" 
This example makes custom 'pie charts' as the markers for a scatter plot 

Thanks to Manuel Metz for the example 
""" 
import math 
import numpy as np 
import matplotlib.pyplot as plt 

# first define the ratios 
r1 = 0.2  # 20% 
r2 = r1 + 0.4 # 40% 

# define some sizes of the plot marker 
markersize = 20 # I changed this line 

# calculate the points of the first pie marker 
# 
# these are just the origin (0,0) + 
# some points on a circle cos,sin 
x = [0] + np.cos(np.linspace(0, 2*math.pi*r1, 10)).tolist() 
y = [0] + np.sin(np.linspace(0, 2*math.pi*r1, 10)).tolist() 

xy1 = list(zip(x, y)) 
s1 = max(max(x), max(y)) 

# ... 
x = [0] + np.cos(np.linspace(2*math.pi*r1, 2*math.pi*r2, 10)).tolist() 
y = [0] + np.sin(np.linspace(2*math.pi*r1, 2*math.pi*r2, 10)).tolist() 
xy2 = list(zip(x, y)) 
s2 = max(max(x), max(y)) 

x = [0] + np.cos(np.linspace(2*math.pi*r2, 2*math.pi, 10)).tolist() 
y = [0] + np.sin(np.linspace(2*math.pi*r2, 2*math.pi, 10)).tolist() 
xy3 = list(zip(x, y)) 
s3 = max(max(x), max(y)) 

fig, ax = plt.subplots() 

# Here's where I made changes 
ax.plot(np.arange(3), np.arange(3), marker=(xy1, 0), 
      ms=markersize, markerfacecolor='blue') # I changed this line 
ax.plot(np.arange(3), np.arange(3), marker=(xy2, 0), 
      ms=markersize, markerfacecolor='green') # I changed this line 
ax.plot(np.arange(3), np.arange(3), marker=(xy3, 0), 
      ms=markersize, markerfacecolor='red') # I changed this line 


plt.margins(0.05) 

plt.show() 

enter image description here

Смежные вопросы