Как позволить пользователю выбрать функцию сюжета (Python / Repl.it)?

Question

Как можно изменить эту программу, чтобы пользователь мог ввести функцию вместо sin(5*x)?

from math import *
import matplotlib as mpl
mpl.use('Agg')
import numpy as np
import matplotlib.pyplot as plt

x = np.arange(-2*pi, 2*pi, 0.1)
y1 = np.sin(5*x)

plt.plot(x, y1)
plt.savefig('graph.png')