Polarplot：要繪制的兩個列表的不同顏色

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])

我的 polarplot 上的所有顏色都相同。 我認為像我一樣為顏色指定浮點數會使它們具有不同的顏色。

例如，我如何將參數中給出的第一個列表繪制為藍色，將第二個繪制為紅色？

Answer 1

您需要創建有效的顏色。

你的colors看起來像這樣[0.005 0.005 0.005 0.2 0.2 0.2 ] 。

例如：

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

用['r' 'r' 'r' 'b' 'b' 'b']創建colors並在你的圖中給出藍色和紅色的點：