在matplotlib中设置变量点大小

Question

我想在散点图中设置变量标记大小.这是我的代码:

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

我导入的文件有三列.列1和2被存储在data[0]和data[1])是(x,y)值,我想的每个点具有相对于柱3(即一个尺寸:data[2])

顺便说一句,我正在使用Canopy IDE.

Answer 1

help(plt.plot) 节目

  markersize or ms: float         

所以它似乎plt.plot不允许markersize成为一个数组.

plt.scatter但是你可以使用:

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

PS.您也可以验证plt.plot的markersize,必须通过搜索'markersize’是一个浮动的官方文档.