simple line chart , scatter diagram and RandomWalk

1,simple line chart

import matplotlib.pyplot as plt

inputvalues=[1,2,3,4,5]
squares=[1,4,9,16,25]

#correct the value
plt.plot(inputvalues,squares,linewidth = 5)

#plt.plot(squares,linewidth = 5) default value from 0

plt.title("Square Number",fontsize = 24)
plt.xlabel("Value",fontsize = 14)
plt.ylabel("Square",fontsize = 14)

plt.tick_params(axis="both",labelsize=14)

plt.show()

2,scatter diagram

import matplotlib.pyplot as plt

#A point
#plt.scatter(2,4,s=20) 

#a serise of points
#~ x_values = [1,2,3,4,5]
#~ y_values = [1,4,9,16,25]

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

#edgecolor default is black
#defalut point color is blue
#c=(0,0,1)

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

#plt.scatter(x_values,y_values,c='red',edgecolor = 'none',s=40)

#set axis range
plt.axis([0,1000,0,1100000])



plt.title("Square Number",fontsize = 24)
plt.xlabel("Value",fontsize = 14)
plt.ylabel("Square",fontsize = 14)

plt.tick_params(axis="both",which="major",labelsize=14)

plt.show()

#save as file
#plt.savefig('squares_plot.png',bbox_inches='tight')

3,RandomWalk()

#random_walk.py

from random import choice

class RandomWalk():

    def __init__(self,num_points=5000):
        self.num_points = num_points

        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)

import matplotlib.pyplot as plt

from random_walk import RandomWalk

#stimulated repeatedly
while True:
    rw=RandomWalk(10000)
    rw.fill_walk()

    #set the window size
    plt.figure(dpi=128,figsize=(10,6))

    #color the points
    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=15)

    #highlight the startpoint and the endpoint
    plt.scatter(0,0,c='green',edgecolor='none',s=100)
    plt.scatter(rw.x_values[-1],rw.y_values[-1],c='red',edgecolor='none',s=100)

    #hide the axis
    plt.axes().get_xaxis().set_visible(False)
    plt.axes().get_yaxis().set_visible(False)


    plt.show()

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

#onece
#~ rw = RandomWalk()
#~ rw.fill_walk()
#~ plt.scatter(rw.x_values,rw.y_values,s=15)

plt.show()