Поларплот: разные цвета для двух списков

Question

Рассмотрим следующий код:

import numpy as np
from numpy import *
from matplotlib.pyplot import *
import matplotlib.pyplot as plt
from mpmath import *
import random

def graphMesure(listeAlpha,listeBeta):
    # Compute areas and colors
    r = np.asarray([1]*len(listeAlpha)+[0.5]*len(listeBeta))
    colors = np.asarray([0.005]*len(listeAlpha)+[0.2]*len(listeBeta))
    area = 200*r**2

    fig = plt.figure()

    ax = fig.add_subplot(111, projection='polar')
    ax.set_ylim([0,1.25])
    c = ax.scatter(listeAlpha+listeBeta, r, c=colors, s=area, cmap='hsv', alpha=1)

graphMesure([0.5,0.2,0.3],[0.7,0.8,0.2])

Все цвета на моем полярном участке одинаковы. Я думал, что указание числа с плавающей точкой для цветов, как я сделал, сделало бы их разных цветов.

Как, например, я могу сделать первый список, заданный в параметре, черным, а второй - красным?

Answer 1

Вам необходимо создать допустимых цветов .

Ваш colors выглядит следующим образом [0.005 0.005 0.005 0.2 0.2 0.2 ].

Например:

colors = np.asarray(['r'] * len(listeAlpha) + ['b'] * len(listeBeta))

создает colors с ['r' 'r' 'r' 'b' 'b' 'b'] и дает синие и красные точки в Ваш участок: