web-dev-qa-db-fra.com

Définir la taille du point variable dans matplotlib

Je veux définir une taille de marqueur variable dans un nuage de points. Ceci est mon code:

import numpy as np
import matplotlib.pyplot as plt

from os import getcwd
from os.path import join, realpath, dirname

mypath = realpath(join(getcwd(), dirname(__file__)))
myfile = 'b34.dat'

data = np.loadtxt(join(mypath,myfile),
     usecols=(1,2,3),
     unpack=True)

fig = plt.figure()
ax1 = fig.add_subplot(111)
ax1.plot(data[0], data[1], 'bo', markersize=data[2], label='the data')
plt.show()

Le fichier que j'importe est composé de trois colonnes. Les colonnes 1 et 2 sont stockées dans data[0] et data[1]) sont les (x,y) valeurs et je veux que chaque point ait une taille relative à la colonne 3 (ie: data[2])

J'utilise le Canopy IDE au fait.

pythonpython-2.7python-3.xmatplotlib
34
Gabriel

help(plt.plot) montre

  markersize or ms: float

il semble donc plt.plot ne permet pas à la taille de marqueur d'être un tableau.

Vous pouvez utiliser plt.scatter toutefois:

ax1.scatter(data[0], data[1], marker='o', c='b', s=data[2], label='the data')

PS Vous pouvez également vérifier que plt.plot _ markersize doit être un float en recherchant "markersize" dans la documentation officielle .

40
unutbu