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