Looking back in my post about energy comparison graph,
is easy to see room for improvement. So in this one I'm sharing a code to make the same plot, but in much more pythonic way. Take a look in the code:
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #Program to plot graph of car, moto and bike power | |
| #Written by Eddie Liberato in 25/10/16 | |
| import numpy as np | |
| import matplotlib.pyplot as plt | |
| import matplotlib.font_manager as font_manager | |
| import matplotlib.patches as mpatches | |
| font_path = '/usr/share/fonts/truetype/strasua/strasua.ttf' | |
| font_prop = font_manager.FontProperties(fname=font_path, size=14) | |
| plt.figure(figsize=(10, 5), dpi=1200) | |
| ax = plt.subplot() | |
| #General parameters | |
| VKmh=60 | |
| Vms=VKmh/3.6 | |
| V=np.linspace(0,Vms,512) | |
| a=0 | |
| g=9.81 | |
| #Defining the main function in SI (Watts) | |
| #m=mass, C_d=drag coef, C_r=rolling resistance coef, I=hill steepness(rise/run) | |
| def potxvel(m, C_d, C_r, I): | |
| f_ine=m*a | |
| f_aer=C_d*V**2 | |
| f_rol=C_r*m*g | |
| f_grav=I*m*g | |
| return V*(f_ine+f_aer+f_rol+f_grav) | |
| #Evaluating funcion with the parameters and coverting to imperial(hp) at same time | |
| Pcarhp=potxvel(m=1000+75, C_d=0.5, C_r=0.03, I=0.02)/735 | |
| Pmotohp=potxvel(m=150+75, C_d=0.4, C_r=0.03, I=0.02)/735 | |
| Pbikehp=potxvel(m=20+75, C_d=0.45, C_r=0.02, I=0.02)/735 | |
| V_label=V*3.6 | |
| #Plotting the results | |
| plt.plot(V_label,Pcarhp, linewidth=3, color='#7E1137', label="Car") | |
| plt.plot(V_label,Pmotohp, linewidth=3, color='#55BA87', label="Moto") | |
| plt.plot(V_label,Pbikehp, linewidth=3, color='#377EB8', label="Bike") | |
| plt.fill_between(V_label,0,Pbikehp,facecolor='#377EB8', alpha=0.8) | |
| plt.fill_between(V_label,Pbikehp,Pmotohp,facecolor='#55BA87', alpha=0.8) | |
| plt.fill_between(V_label,Pmotohp,Pcarhp,facecolor='#7E1137', alpha=0.8) | |
| ax.grid (True) | |
| plt.xlim(V_label.min(),V_label.max()) | |
| for label in (ax.get_xticklabels() + ax.get_yticklabels()): | |
| label.set_fontproperties(font_prop) | |
| label.set_fontsize(13) | |
| plt.title("Car VS Moto VS Bike", fontproperties=font_prop,size=16, verticalalignment='bottom') | |
| plt.xlabel("Speed (Km/h)", fontproperties=font_prop) | |
| plt.ylabel("Power (Hp) ", fontproperties=font_prop) | |
| plt.legend(loc='upper left', prop=font_prop) | |
| plt.savefig('PotXvel.svg', format='svg', dpi=1200) |
In the first time I refused using "def" to define the function, thinking this would be overkill for a simple script, and would affect the readability. I was wrong, now the code bloat is reduced, and the readability is even better. The way python works, permitting that you define the parameter value inside an array (look at lines 31, 32, 33), keeps the readability, and also makes the order of the parameters in the array unimportant. Sweeeeeet.
