Need to code this wave function in Python

Assume a simple sinusoidal wave of the form n(x, t)-Acos(kx - ot) where ,, is the wave elevation (displacement above mean water level in y-direction), 'A, is the wave amplitude, 'K is the wavenumber, ''is the circular frequency, 'X is the distance along the x-direction and is time. Notice ,, is 2D (matrix) depending on both X and 't' The wavenumber k-2; where'' is the wavelength. 2 The circular frequency where 'T' is the Time Period of the wave. (kxt) is the Phase Angle For the following parameters: A 1 meter; Time Period 10 seconds; Wavelength-156 meters; No of waves 10 Plot the following on each figure: Wave elevation versus time for a given locations in space: x [0, X, , X, 1] * wavelength Wave elevation versus space x for a given instants in time: ts [0, X, , X , 1] * Time Period a. b. HINT: (x, t) can be computed as a Matrix with X in rows and in columns or vice versa Make sure all plots are labeled. Below is how your plots could be formatted