pywt

1
import matplotlib.pyplot as plt
2
import numpy as np
3

4
import pywt
5

6
s = np.linspace(-4, 4, 1000)
7

8
s_soft = pywt.threshold(s, value=0.5, mode='soft')
9
s_hard = pywt.threshold(s, value=0.5, mode='hard')
10
s_garrote = pywt.threshold(s, value=0.5, mode='garrote')
11
s_firm1 = pywt.threshold_firm(s, value_low=0.5, value_high=1)
12
s_firm2 = pywt.threshold_firm(s, value_low=0.5, value_high=2)
13
s_firm3 = pywt.threshold_firm(s, value_low=0.5, value_high=4)
14

15
fig, ax = plt.subplots(1, 2, figsize=(10, 4))
16
ax[0].plot(s, s_soft)
17
ax[0].plot(s, s_hard)
18
ax[0].plot(s, s_garrote)
19
ax[0].legend(['soft (0.5)', 'hard (0.5)', 'non-neg. garrote (0.5)'])
20
ax[0].set_xlabel('input value')
21
ax[0].set_ylabel('thresholded value')
22

23
ax[1].plot(s, s_soft)
24
ax[1].plot(s, s_hard)
25
ax[1].plot(s, s_firm1)
26
ax[1].plot(s, s_firm2)
27
ax[1].plot(s, s_firm3)
28
ax[1].legend(['soft (0.5)', 'hard (0.5)', 'firm(0.5, 1)', 'firm(0.5, 2)',
29
              'firm(0.5, 4)'])
30
ax[1].set_xlabel('input value')
31
ax[1].set_ylabel('thresholded value')
32
plt.show()
33