Python数据可视化

来源：互联网发布：太原知达常青藤好进吗编辑：程序博客网时间：2024/04/30 13:14

首先show一个示例：

from numpy import *import matplotlibimport matplotlib.pyplot as pltfr = open("datingTestSet.txt")lines = fr.readlines()n = len(lines)datingDataset = zeros((n,3))datingLabels = []i = 0for line in lines:    fdata = line.strip().split('\t')    datingDataset[i,0:3] = fdata[0:3]    datingLabels.append(int(fdata[-1]))    i = i + 1 fig = plt.figure()ax = fig.add_subplot(111)ax.scatter(datingDataset[:,1],datingDataset[:,2],20.0*array(datingLabels),20.0*array(datingLabels))plt.show()

注：datingTestSet.txt是一个样本集合，每行代表一个4维的样本，前三维是样本的特征属性，第四维是样本所属的类别标签（共三类：1,2,3）

在可视化程序中，只用了每个数据的第二维和第三维

ax.scatter(datingDataset[:,1],datingDataset[:,2],20.0*array(datingLabels),20.0*array(datingLabels))

其中，20.0*array(datingLabels),20.0*array(datingLabels)是用类别标签标表示不同类别点的颜色和大小，结果如图：

0 0