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()

输出结果：

  

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