python matplotlib 2D 散点图的制作（输入二维坐标） |
您所在的位置：网站首页 › 二维平面图怎么画出一个原度 › python matplotlib 2D 散点图的制作（输入二维坐标） |
python matplotlib 2D 散点图的制作（输入二维坐标）
|
相信好多童鞋遇到这样一种情况，用matplotlib想直接画出一个二维平面上的离散点，如果用一个1000*100的数组来表示二位平面的坐标，其值为0或者1，1表示存在，0表示不存在。用到python的scatter函数直接操作，发现比较困难。尤其不知道什么import file2matrix的时候... 下面我具体探讨下： scatter函数如下：
查看参数发现其中的X,Y 分别坐标点，与我们1000*100数组中的xy不是同一个概念。那么首先需要进行坐标转换，把这样一个数组转换为100000*2的数组，其中[n,0]存放x坐标，[n,1]存放y坐标。具体如何转换呢，我就自己贴代码了，您一看就明白： # -*- coding: utf-8 -*- #coding=utf-8 #导入必要的模块 from numpy import * import matplotlib.pyplot as plt #产生测试数据 ''' x = np.arange(1,10) y = x ''' data_spa=zeros((1000,100)) for i in range(0,100,1): data_spa[200,i]=1 for i in range(0,1000,1): data_spa[i,5]=1 data_spa_1=zeros((100000,2)) for x in range(0,1000,1): for y in range(0,100,1): if data_spa[x,y]==1: data_spa_1[100*x+y,0]=x data_spa_1[100*x+y,1]=y fig = plt.figure() ax1 = fig.add_subplot(111) #设置标题 ax1.set_title('Scatter Plot') #设置X轴标签 plt.xlabel('X') #设置Y轴标签 plt.ylabel('Y') #画散点图 ax1.scatter(x=data_spa_1[:,0],y=data_spa_1[:,1],c = 'r',marker = 'o') #设置图标 plt.legend('x1') #显示所画的图 plt.show()输出结果：
其中可以看见我们给定的两条红色直线（其实是由散点组成），欢迎交流。 |
【本文地址】