#!usr/bin/env python
#encoding:utf-8
"""
__Author__:沂水寒城
功能:折线图、散点图测试
"""
import random
import matplotlib
import matplotlib.pyplot as plt
def list2mat(data_list,w):
"""
切片、转置
"""
mat=[] res=[] for i in range(0,len(data_list)-w+1,w):
mat.append(data_list[i:i+w])
for i in range(len(mat[0])):
one_list=[] for j in range(len(mat)):
one_list.append(mat[j][i])
res.append(one_list)
return res
def draw_pic_test():
"""
作图
"""
data_list=[] for i in range(100):
data_list.append(random.randint(2,150))
month_list=range(1,11,1)
mat=list2mat(data_list,w=10)
for one_list in mat:
one_list=[int(one) for one in one_list] plt.plot(month_list,one_list,"x-",label="test_zhexian")
plt.savefig("test_zhexian.png")
plt.close()
for one_list in mat:
one_list=[int(one) for one in one_list] plt.scatter(month_list,one_list,marker="x",label="test_sandian",s=30)
plt.savefig("test_sandian.png")
plt.close()
if __name__ == "__main__":
draw_pic_test()