在 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 不允许标记大小是一个数组。

不过你可以使用plt.scatter：

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

---

PS。您还可以通过在the official documentation 中搜索“markersize”来验证plt.plot 的markersize 必须是浮点数。