编程语言
首页 > 编程语言> > Python散点图:标记样式的条件?

Python散点图:标记样式的条件?

作者:互联网

我有一个数据集,我希望用matplotlib绘制为散点图,以及一个相同大小的矢量,用于对数据点进行分类和标记(离散地,例如从0到3).我想为不同的标签使用不同的标记(例如,’x’代表0,’o’代表1,依此类推).我怎样才能优雅地解决这个问题?我很确定我只是错过了一些东西,但是并没有真正找到它,而且我的天真方法到目前为止都失败了……

解决方法:

如何迭代这样的所有标记:

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(100)
y = np.random.rand(100)
category = np.random.random_integers(0, 3, 100)

markers = ['s', 'o', 'h', '+']
for k, m in enumerate(markers):
    i = (category == k)
    plt.scatter(x[i], y[i], marker=m)

plt.show()

标签:scatter-plot,python,matplotlib
来源: https://codeday.me/bug/20190725/1534368.html