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