最近想要完成一个小的画图程序。因为python中的matplotlib很方便,所以完成一个python画散点图的小程序。
数据存在csv文件里面,使用csv包进行读取,画图。
#导入必要的模块
import numpy as np
import matplotlib.pyplot as plt
import csv
#产生测试数据
ii = 0
filename = 'result.csv'
x = []
y = []
with open(filename) as f:
reader = csv.reader(f)
item = reader
for i in reader:
x .append (float(i[0]))
y .append (float(i[1]))
fig = plt.figure()
ax1 = fig.add_subplot(111)
#设置标题
ax1.set_title('Scatter Plot')
#设置X轴标签
plt.xlabel('X')
#设置Y轴标签
plt.ylabel('Y')
#画散点图
ax1.scatter(x,y,c = 'r',marker = 'o')
#设置图标
plt.legend('x1')
#显示所画的图
plt.show()
画图如下。