# 0.4 Plotting Signals

### Plot the Signal

Let's display the signal. To make calling these multiple times easier, let's write a function. Further, to be able to save the results, let's add a conditional statement to allow saving the result.

```python
def plot_signal(x, xaxis_label=None, yaxis_label=None, save=None):
    plt.plot(x)
    plt.xlabel(xaxis_label)
    plt.ylabel(yaxis_label)
    
    if save:
        plt.savefig(save + '.pdf', format='pdf')
    
    return

plot_signal(s, xaxis_label=r'$n$', yaxis_label=r'$s[n]$', save='input')
```

Feel free to add more functionality to the plotting function. Possible explorations are to check to add `legends` , `titles`, `linestyles`, `linewidths`, `plot colours`. See Matplotlib [documentation](https://matplotlib.org/stable/contents.html) for more.
