python - 来自 pandas 数据框的不同大小、标记和颜色的散点图

Question

我正在尝试为每个点绘制速度超过米的散点图，其中标记表示不同的类型，大小表示不同的重量，颜色表示一个点超过 10 分钟刻度的时间。但是，到目前为止我只能按大小绘制。

非常感谢任何帮助。

x = {'speed': [10, 15, 20, 18, 19], 'meters' : [122, 150, 190, 230, 300], 'type': ['phone', 'phone', 'gps', 'gps', 'car'], 'weight': [0.2, 0.3, 0.1, 0.85, 0.0], 'old': [1, 2, 4, 5, 8]}

m = pd.DataFrame(x)

plt.scatter(m.meters, m.speed, s = 30* m.weight)

mkr_dict = {'gps': 'x', 'phone': '+', 'car': 'o'}

   meters  speed   type  weight  old
0     122     10  phone    0.20    1
1     150     15  phone    0.30    2
2     190     20    gps    0.10    4
3     230     18    gps    0.85    5
4     300     19    car    0.00    8

更新的问题:

我正在尝试将 colorbar 添加到基于旧的色标。当我针对整个数据集进行绘图但在尝试为每个子集添加标记后失败时它起作用了。有什么想法吗？

plt.scatter(m.meters, m.speed, s = 30* m.weight, c=m.old)
cbar = plt.colorbar(ticks = [0, 5, 10])
cbar.ax.set_yticklabels(['New','5mins', '10mins'])

TypeError:您必须先为可映射设置 set_array

Answer 1

scatter 一次只能做一种标记，所以你必须分别绘制不同的类型。幸运的是，pandas 让这一切变得简单:

import matplotlib.pyplot as plt
import pandas as pd
x = {'speed': [10, 15, 20, 18, 19],
     'meters' : [122, 150, 190, 230, 300],
     'type': ['phone', 'phone', 'gps', 'gps', 'car'],
     'weight': [0.2, 0.3, 0.1, 0.85, 0.0],
     'old': [1, 2, 4, 5, 8]}

m = pd.DataFrame(x)
mkr_dict = {'gps': 'x', 'phone': '+', 'car': 'o'}
for kind in mkr_dict:
    d = m[m.type==kind]
    plt.scatter(d.meters, d.speed, 
                s = 100* d.weight, 
                c = d.old, 
                marker = mkr_dict[kind])
plt.show()

.... 车在哪儿？好吧，原始测试数据中的权重是 0.0，而我们使用的是标记大小的权重，所以:看不到它。