python - 在 python plt.plot 中使用系列作为标记大小

Question

是否可以使用数据框中的列来缩放 matplotlib 中的标记大小？当我执行以下操作时，我不断收到有关使用系列的错误消息。

import pandas as pd
import matplotlib.pyplot as plt

my_dict = {'Vx': [16,25,85,45], 'r': [1315,5135,8444,1542], 'ms': [10,50,100, 25]}
df= pd.DataFrame(my_dict)
fig, ax = plt.subplots(1, 1, figsize=(20, 10))
ax.plot(df.Vx, df.r, '.', markersize= df.ms)

当我运行时

ValueError: setting an array element with a sequence.

我猜它不喜欢我向标记器提供系列的事实，但必须有一种方法让它工作......

Answer 1

使用plt.scatter而不是 plt.plot。 Scatter 允许您使用元组或列表指定大小 s 以及点的颜色 c。

import pandas as pd
import matplotlib.pyplot as plt

my_dict = {'Vx': [16,25,85,45], 'r': [1315,5135,8444,1542], 'ms': [10,50,100, 25]}
df= pd.DataFrame(my_dict)
fig, ax = plt.subplots(1, 1, figsize=(20, 10))
ax.scatter(df.Vx, df.r, s= df.ms)
plt.show()