python画散点图 不同属性赋予不同颜色

问题描述

数据结构：

其中cluster_db一共四个值（-1、0、1、2）

需要将经纬度作为x y轴绘制散点图，并且不同的cluster_db需要赋予不同的颜色显示

解决方法

1.内置颜色

首先导入数据

import pandas as pd

beer = pd.read_csv('E:\\1\\20201112-Mobility diversity\\test.csv')
df = pd.DataFrame(beer)

接着通过matplotlib.pyplot包绘图

import matplotlib.pyplot as plt

df.plot.scatter('LAT', 'LON', c='cluster_db', colormap='jet')
plt.title('result1')
plt.show()

结果：

2.自定义颜色

同样先导入数据

import pandas as pd

beer = pd.read_csv('E:\\1\\20201112-Mobility diversity\\test.csv')

定义一个空数组和颜色数组（我定义了5个）

col = []
colors = ['#FF0000', '#FFA500', '#FFFF00', '#00FF00', '#228B22']

通过循环将不同的cluster_db值对应的颜色加到col数组中，其中label就是cluster_db数组

for i in range(0, len(labels)):
    col.append(colors[labels[i]])

最后绘图

import matplotlib.pyplot as plt

df.plot.scatter('LAT', 'LON',  c=col, colormap='jet')
plt.title('result2')
plt.show()

结果：