TTT4120/assignments/01/conv.py

32 lines
515 B
Python

import numpy as np
x = [1,2,3]
h_1 = [1,1,1]
h2Length = 10
h_2 = [(0.9**n) for n in range(h2Length + 1)]
print(h_2)
y_1 = np.convolve(x,h_1)
y_2 = np.convolve(y_1,h_2)
y_1h2 = np.convolve(x,h_2)
print(y_1)
print(y_2)
print(y_1h2)
def GenStem(arr, padding = 2):
print("n\txn")
for i in range(padding):
print(str(i-padding) + "\t0")
for i in range(len(arr)):
print(str(i) + "\t" + str(arr[i]))
for i in range(padding):
print(str(len(arr) + i) + "\t0")
GenStem(y_1h2)