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]=1for i in range(0,1000,1):    data_spa[i,5]=1data_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]=yfig = 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()

输出结果：

  

其中可以看见我们给定的两条红色直线（其实是由散点组成），欢迎交流。