Python学习笔记-matplotlib

        一、安装matplotlib

        利用pip安装，在命令行中输入

pip install matplotlib

        pip就会自动下载matplotlib。下载完成后进入python的交互模式，输入

import matplotlib

　　如果没有出现错误信息，表明安装成功。

        二、绘制折线图和散点图

import matplotlib.pyplot as plt

input_values=[1,2,3,4,5]
squares=[1,4,9,16,25]
plt.plot(input_values,squares,linewidth=5)

#set title and label
plt.title('Square Numbers',fontsize=24)
plt.xlabel('Value',fontsize=14)
plt.ylabel('Square of Value',fontsize=14)

#scale size
plt.tick_params(axis='both',labelsize=14)
plt.show()

        绘制散点图，并让点的颜色渐变

import matplotlib.pyplot as plt

x_values=list(range(1,1001))
y_values=[x**2 for x in x_values]

plt.scatter(x_values,y_values,s=40)
#set title and label
plt.title('Square Numbers',fontsize=24)
plt.xlabel('Value',fontsize=14)
plt.ylabel('Square of value',fontsize=14)

#set scale
plt.tick_params(axis='both',which='major',labelsize=14)

plt.scatter(x_values,y_values,c=y_values,cmap=plt.cm.Blues,edgecolor='none',s=40)

plt.axis([0,1100,0,1100000])

plt.show()
# ~ plt.savefig('squares_plot.png',bbox_inches='tight')

        三、随机漫步

        1、创建RandomWalk()类

from random import choice

class RandomWalk():
    
    def __init__(self,num_points=5000):
        self.num_points=num_points
        
        #all random walk from (0,0)
        self.x_values=[0]
        self.y_values=[0]
        
    def fill_walk(self):
        
        while len(self.x_values)<self.num_points:
            x_direction=choice([1,-1])
            x_distance=choice([0,1,2,3,4])
            x_step=x_direction*x_distance
            
            y_direction=choice([1,-1])
            y_distance=choice([0,1,2,3,4])
            y_step=y_direction*y_distance
            
            if x_step==0 and y_step==0:
                continue
                
            next_x=self.x_values[-1]+x_step
            next_y=self.y_values[-1]+y_step
            
            self.x_values.append(next_x)
            self.y_values.append(next_y)

        2、绘制随机漫步图

import matplotlib.pyplot as plt

from random_walk import RandomWalk

while True:
    rw=RandomWalk(50000)
    rw.fill_walk()
    
    point_numbers=list(range(rw.num_points))
    plt.scatter(rw.x_values,rw.y_values,c=point_numbers,cmap=plt.cm.Blues,
        edgecolor='none',s=1)
    
    plt.scatter(0,0,c='green',edgecolors='none',s=100)
    plt.scatter(rw.x_values[-1],rw.y_values[-1],c='red',edgecolors='none',
        s=100)
    
    plt.axes().get_xaxis().set_visible(False)
    plt.axes().get_yaxis().set_visible(False)
    
    plt.show()

    keep_running=input('Make another walk?(y/n): ')
    if keep_running=='n':
        break

        待解决的问题，matplotlib不支持中文和中文注释。