Course source code: https://codistwa.com/guides/functions-in-math. More courses on https://codistwa.com

functions-in-math.py

# ============================================================
# Definition of fonction
# ============================================================

def area(x):
    return 2 * x

print(area(4))

x = np.linspace(-6,6)
f = x*2

plt.plot(x,f,label='f')

plt.axis([0,6,0,6])
plt.grid()
plt.legend()

plt.title('Function',fontsize=12)
plt.savefig('function', bbox_inches='tight')