欢迎使用CSDN-markdown编辑器

昨天第一次成功独立写了个自己的python程序，要把其中的细节都记下来。

一、总览

这次的需求是：

1) 给定文件夹中有一批数据文件；

图1

2) 轮流读取每个文件；
3) 对每个文件要读取每一行；

图1

4) 第一列是id，第二三列分别是x、y坐标；
5) 给每个文件用python画个图，标出该文件中所有id所表示的object的具体位置（第一个id还有radius，要画出它的圈圈），然后将此图存成png文件如下：

图1

6) 批量生成每个数据文件对应的png文件

最后的整个程序如下：

import osimport matplotlib.pyplot as pltfrom matplotlib.ticker import MultipleLocatorSRC_DIR = "E:\\0-Some--WorkSpaces\\MyPython\\coor_map\\src"DST_DIR = "E:\\0-Some--WorkSpaces\\MyPython\\coor_map\\dst"for parent, dirs, files in os.walk(SRC_DIR):    for f in files:        fig = plt.figure()        ax = fig.add_subplot(111)        xmajorLocator = MultipleLocator(0.05)        xminorLocator = MultipleLocator(0.01)        ymajorLocator = MultipleLocator(0.05)        yminorLocator = MultipleLocator(0.01)       ax.xaxis.set_major_locator(xmajorLocator)       ax.yaxis.set_major_locator(ymajorLocator)        ax.xaxis.set_minor_locator(xminorLocator)        ax.yaxis.set_minor_locator(yminorLocator)        ax.xaxis.grid(True, which='major')        ax.yaxis.grid(True, which='minor')        file = os.path.join(parent, f)        with open(file) as srcfile:            lines = srcfile.readlines()            count = -1            for line in lines:                if count == -1:                    cur_id = int(line.split('\t')[0])                    cur_x = float(line.split('\t')[1])                    cur_y = float(line.split('\t')[2])                    cur_rad = float(line.split('\t')[3])                    color = 'red'                    scale = 20                    plt.scatter(cur_x, cur_y, color='red', s=scale, label='curobj', alpha=0.6, edgecolors='white')                    circ = plt.Circle((cur_x, cur_y), cur_rad, color='r', alpha=0.3)                    ax.add_patch(circ)                    # add annotation                    ax.annotate("su=" + str(cur_rad),                                xy=(cur_x, cur_y), xycoords='data',                                xytext=(cur_x+0.1, cur_y-0.1), textcoords='data',                                arrowprops=dict(arrowstyle="->",                                                connectionstyle="arc3"),                                )                else:                    fri_id = line.split('\t')[0]                    fri_x = line.split('\t')[1]                    fri_y = line.split('\t')[2]                    color = 'blue'                    scale = 20                    if count == 0:                        plt.scatter(fri_x, fri_y, color='blue', s=scale, label='friobj', alpha=0.6, edgecolors='white')                    else:                        plt.scatter(fri_x, fri_y, color='blue', s=scale, alpha=0.6, edgecolors='white')                count += 1            plt.title('')            plt.xlabel('x')            plt.ylabel('y')            plt.legend()            plt.grid(True)            dstfile = DST_DIR + os.path.sep + f + '.png'            plt.savefig(dstfile)            # plt.show()

二、画图

使用matplotlib当然首选看它的官方文档咯：
http://matplotlib.org/api/index.html

1、画散点

主要用到了scatter()函数，而如何画散点则主要是参考了大神“烂笔头”的http://note4code.com/2015/03/30/%E4%BD%BF%E7%94%A8matplotlib%E7%BB%98%E5%88%B6%E6%95%A3%E7%82%B9%E5%9B%BE/
对应的上面的代码部分如下：

color = 'red'scale = 20plt.scatter(cur_x, cur_y, c=color, s=scale, label='curobj', alpha=0.6, edgecolors='white')

前两个参数是该散点的x、y坐标，第三个参数是颜色，第四个参数是该点的大小，第五个参数是写在图例里的标签，第六个参数alpha好像是调亮度的，然后最后一个参数是边上的颜色

2、画圈圈

主要用的是plt.Circle()函数：
http://matplotlib.org/examples/shapes_and_collections/artist_reference.html
当然，有个非官方的也很好，专门讲画形状的：
https://nickcharlton.net/posts/drawing-animating-shapes-matplotlib.html
我用到的代码是：

circ = plt.Circle((cur_x, cur_y), cur_rad, color='r', alpha=0.3)ax.add_patch(circ)

同样，指定中心坐标、半径，以及颜色等等

3、画注释

这里我主要参考的matplotlib的官方文档：http://matplotlib.org/users/annotations_intro.html，官方文档有各种 source code 以及结果图等可以尽情利用，真是好极了！
我的代码如下：

# add annotationax.annotate("su=" + str(cur_rad),            xy=(cur_x, cur_y), xycoords='data',            xytext=(cur_x+0.1, cur_y-0.1),             textcoords='data',             arrowprops=dict(arrowstyle="->",            connectionstyle="arc3"),)

参数列表里，最后竟然留了个逗号，也是奇怪……

4、标刻度

因为有的时候坐标轴的刻度要是不对的话，那些点和圈圈都显示不出来，还会以为是代码出错了呢

设刻度主要是参考了一篇中文博客：http://blog.csdn.net/rumswell/article/details/6571005
我的代码如下：

xmajorLocator = MultipleLocator(0.05)xminorLocator = MultipleLocator(0.01)ymajorLocator = MultipleLocator(0.05)yminorLocator = MultipleLocator(0.01)       ax.xaxis.set_major_locator(xmajorLocator)       ax.yaxis.set_major_locator(ymajorLocator)        ax.xaxis.set_minor_locator(xminorLocator)        ax.yaxis.set_minor_locator(yminorLocator)ax.xaxis.grid(True, which='major')ax.yaxis.grid(True, which='minor')

一二行是定义x轴的‘主刻度’和‘次刻度’，三四行是定义y轴的‘主刻度’和‘次刻度’，后面的就是真正将四个刻度设置生效

三、读写文件

我这里是要遍历某目录下的所有文件。所以，
首先，把要读取的源文件夹和要写入的目标文件夹定义好：

SRC_DIR = "E:\\0-Some--WorkSpaces\\MyPython\\coor_map\\src"DST_DIR = "E:\\0-Some--WorkSpaces\\MyPython\\coor_map\\dst"

当然，一个较好的习惯是还生成 log 文件，就定义一个 LOG_DIR。
然后，遍历源目录下的所有文件，用for循环就可以对每个文件进行处理了，之后打开一个源文件和一个目标文件，分别进行读写：

file = os.path.join(parent, f)        with open(file) as srcfile:            lines = srcfile.readlines()

我这里没有打开目标文件，而是在后面定义的：

dstfile = DST_DIR + os.path.sep + f + '.png'plt.savefig(dstfile)

因为这里是要保存png的图片，所以plt.savefig(dstfile)中的‘dstfile’得是一个字符串，而不是像写文本文件一样，用dstfile.write()就可以了，所以，在上面代码中的 with 后面打开‘open(DST_DIR + os.path.sep + f, “w+”) as dstFile’是没用滴

大概记录到这儿