File indexing completed on 2024-05-19 03:49:04
0001 #!/usr/bin/env python 0002 ########################################################################### 0003 # File : nsl_smooth_check.h 0004 # Project : LabPlot 0005 # Description : NSL smooth functions 0006 # -------------------------------------------------------------------- 0007 # SPDX-FileCopyrightText: 2016 Stefan Gerlach <stefan.gerlach@uni.kn> 0008 # 0009 ########################################################################### 0010 0011 ########################################################################### 0012 # # 0013 # SPDX-License-Identifier: GPL-2.0-or-later 0014 # # 0015 ########################################################################### 0016 0017 from scipy.signal import savgol_filter 0018 import numpy as np 0019 x = np.array([2, 2, 5, 2, 1, 0, 1, 4, 9]) 0020 0021 m=5 0022 order=2 0023 np.set_printoptions(precision=4) 0024 print x 0025 print "mode:interp" 0026 print savgol_filter(x, m, order, mode='interp') 0027 print "mode:mirror" 0028 print savgol_filter(x, m, order, mode='mirror') 0029 print "mode:nearest" 0030 print savgol_filter(x, m, order, mode='nearest') 0031 print "mode:constant" 0032 print savgol_filter(x, m, order, mode='constant') 0033 print "mode:wrap" 0034 print savgol_filter(x, m, order, mode='wrap')