import numpy as np import matplotlib.pyplot as plt def main(): #x_axis = np.linspace(-5, 5, 100) f_s = 3 T_s = 1/f_s x_axis = np.arange(-5, 5, T_s) x_axis_2 = np.linspace(-5, 5, len(x_axis)) y_axis = x_axis**2 plt.stem(x_axis, y_axis) plt.show() if __name__ == "__main__": main()