Matplotlib nuage de points; la couleur en fonction d'une troisième variable

Je veux faire un nuage de points (à l'aide de matplotlib) où les points sont à l'ombre selon une troisième variable. J'ai de très près avec ceci:

plt.scatter(w, M, c=p, marker='s')

où w et M sont les points de données et p est la variable que je veux de l'ombre à l'égard de.

Cependant, je veux le faire en niveaux de gris plutôt que de la couleur. Quelqu'un peut-il aider?

InformationsquelleAutor Thomas Collett | 2011-11-20
  1. 137

    Il n'est pas nécessaire de régler manuellement les couleurs. Au lieu de cela, spécifier une palette en niveaux de gris...

    import numpy as np
import matplotlib.pyplot as plt

# Generate data...
x = np.random.random(10)
y = np.random.random(10)

# Plot...
plt.scatter(x, y, c=y, s=500)
plt.gray()

plt.show()

    Matplotlib nuage de points; la couleur en fonction d'une troisième variable

    Ou, si vous préférez une large éventail de la carte des couleurs, vous pouvez également spécifier le cmap kwarg à scatter. Pour utiliser la version inversée de l'un de ces, il suffit de spécifier le "_r" version de l'un d'eux. E. g. gray_r au lieu de gray. Il existe plusieurs niveaux de gris colormaps pré-faites (par exemple,gray, gist_yarg, binary, etc). 

    import matplotlib.pyplot as plt
import numpy as np

# Generate data...
x = np.random.random(10)
y = np.random.random(10)

plt.scatter(x, y, c=y, s=500, cmap='gray')
plt.show()
    • Merci! Est-il de toute façon à dessiner les contours autour de ces points contenant une certaine quantité de la masse totale?
    • mpl.cm est également disponible directement comme plt.cm.
    • Collet: Si vous voulez dessiner les contours, vous auriez à interpoler le formulaire de données de points d'une matrice 2D, puis de la parcelle que l'utilisation de plt.contour() ou plt.contourf() -- mais c'est une autre question
    • Comment pouvez-vous ajouter l'étiquette dans la légende? Dans ce cas, avec des nombres continus, je suppose que vous utilisez la barre de couleur. Quid dans le cas de valeurs discrètes? par exemple, pourrais-je ajouter 3 étiquettes dans la légende pour mes trois catégories?
    InformationsquelleAutor Joe Kington
  2. 25

    Dans matplotlib gris peut être donnée comme une chaîne de caractères en une valeur numérique entre 0 et 1.

    Par exemple c = '0.1'

    Ensuite, vous pouvez convertir votre troisième variable à une valeur à l'intérieur de cette gamme à l'utilisation et à la couleur de vos points.

    Dans l'exemple suivant, j'ai utilisé la position y du point que la valeur qui détermine la couleur: 

    from matplotlib import pyplot as plt

x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [125, 32, 54, 253, 67, 87, 233, 56, 67]

color = [str(item/255.) for item in y]

plt.scatter(x, y, s=500, c=color)

plt.show()

    Matplotlib nuage de points; la couleur en fonction d'une troisième variable

    • Pour l'enregistrement, vous pouvez appliquer ALPHA comme une palette de couleurs si vous fournissez une (n,4) en forme de tableau comme c argument, avec des valeurs alpha dans la quatrième colonne. Si le troisième variable Z, avec shape=(n,1), puis colors = numpy.hstack((numpy.zeros_like(z), numpy.zeros_like(z), numpy.ones_like(z), z/z.max())) donne un très joli effet (bien sûr, il peut être modifié).
    • J'obtiens une erreur: 'longueur de rgba séquence doit être de 3 ou 4'
    • Je soupçonne que vous appelez plt.l'intrigue plutôt que de plt.scatter comme dans l'exemple.
    • comment ajouter une barre de couleur à cette intrigue?
    • vous pouvez en faire une nouvelle question
    • va le faire

    InformationsquelleAutor joaquin
  3. 5

    Parfois, vous devrez peut-être parcelle couleur précisément en fonction de la valeur x cas. Par exemple, vous pouvez avoir un dataframe avec 3 types de variables et de certains points de données. Et vous voulez faire à la suite,

    • Parcelle de points correspondant à la Physique de la variable " A " en ROUGE.
    • Parcelle de points correspondant à la variable Physique " B " en BLEU.
    • Parcelle de points correspondant à la Physique de la variable " C " en VERT.

    Dans ce cas, vous pouvez avoir à écrire de courts fonction de mapper les valeurs de x correspondant à des noms de couleurs, comme une liste et ensuite transmettre cette liste à la plt.scatter commande.

    x=['A','B','B','C','A','B']
y=[15,30,25,18,22,13]

# Function to map the colors as a list from the input list of x variables
def pltcolor(lst):
    cols=[]
    for l in lst:
        if l=='A':
            cols.append('red')
        elif l=='B':
            cols.append('blue')
        else:
            cols.append('green')
    return cols
# Create the colors list using the function above
cols=pltcolor(x)

plt.scatter(x=x,y=y,s=500,c=cols) #Pass on the list created by the function here
plt.grid(True)
plt.show()

    Matplotlib nuage de points; la couleur en fonction d'une troisième variable

    InformationsquelleAutor Tirtha