# -*- coding: utf-8 -*-
 
import numpy as np
import pywt
 
data = np.linspace(1, 4, 7)
 
# pywt.threshold方法講解：
#    pywt.threshold（data，value，mode ='soft'，substitute = 0 ）
#    data：數(shù)據集，value：閾值，mode：比較模式默認soft，substitute：替代值，默認0，float類型
 
#data: [ 1. 1.5 2. 2.5 3. 3.5 4. ]
#output：[ 6. 6. 0. 0.5 1. 1.5 2. ]
#soft 因為data中1小于2，所以使用6替換，因為data中第二個1.5小于2也被替換，2不小于2所以使用當前值減去2，，2.5大于2，所以2.5-2=0.5.....
print "---------------------soft:絕對值-------------------------"
print pywt.threshold(data, 2, 'soft',6)
 
print "---------------------hard:絕對值-------------------------"
 
#data: [ 1. 1.5 2. 2.5 3. 3.5 4. ]
#hard data中絕對值小于閾值2的替換為6，大于2的不替換
print pywt.threshold(data, 2, 'hard',6)
 
print "---------------------greater-------------------------"
 
#data: [ 1. 1.5 2. 2.5 3. 3.5 4. ]
#data中數(shù)值小于閾值的替換為6，大于等于的不替換
print pywt.threshold(data, 2, 'greater',6)
print "---------------------less-------------------------"
print data
#data: [ 1. 1.5 2. 2.5 3. 3.5 4. ]
#data中數(shù)值大于閾值的，替換為6
print pywt.threshold(data, 2, 'less',6)