首页
学习
活动
专区
圈层
工具
发布
社区首页 >专栏 >python 小波包分解_小波分解示意图

python 小波包分解_小波分解示意图

作者头像
全栈程序员站长
发布2022-11-17 10:18:22
发布2022-11-17 10:18:22
2.1K0
举报

大家好,又见面了,我是你们的朋友全栈君。

由于最近正好在学习用python进行小波分解,看的英文的pywt库的各种属性和方法及其使用示例,在这里记录下来,方便以后查阅,前面的小波分解部分忘了记录了,就只能从小波包分解开始了。 小波包: 首先导入pywt库:

代码语言:javascript
复制
>>> import pywt

一、创建小波包结构: 接下来我们实例化一个小波包对象:

代码语言:javascript
复制
>>> x = [1, 2, 3, 4, 5, 6, 7, 8]
>>> wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric')

输入数据和分解系数(细节系数和逼近系数)都可以通过WaveletPacket.data得到:

代码语言:javascript
复制
>>> print(wp.data)
[1, 2, 3, 4, 5, 6, 7, 8]

小波包树的节点由路径标识。标识根节点的路径是’ ‘,根节点的分解层数为0。

代码语言:javascript
复制
>>> print(repr(wp.path))
''
>>> print(wp.level)
0

关于最大分解层数,如果构造函数中没有指定参数,则自动计算。

代码语言:javascript
复制
>>> print(wp['ad'].maxlevel)
3

二、遍历小波包树 获取子节点:

代码语言:javascript
复制
>>> x = [1, 2, 3, 4, 5, 6, 7, 8]
>>> wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric')

首先要先检查最大分解层数:

代码语言:javascript
复制
>>> print(wp.maxlevel)
3

下面开始获取小波包树的子节点:

  • 1st level
代码语言:javascript
复制
>>> print(wp['a'].data)
[  2.12132034   4.94974747   7.77817459  10.60660172]
>>> print(wp['a'].path)
a
  • 2nd level
代码语言:javascript
复制
>>> print(wp['aa'].data)
[  5.  13.]
>>> print(wp['aa'].path)
aa
  • 3rd level
代码语言:javascript
复制
>>> print(wp['aaa'].data)
[ 12.72792206]
>>> print(wp['aaa'].path)
aaa

以上,我们已经达到了最大分解层数,如果接着往下遍历会产生一个索引错误。

代码语言:javascript
复制
>>> print(wp['aaaa'].data)
Traceback (most recent call last):
...
IndexError: Path length is out of range.

现在我们尝试一下输入错误的路径:

代码语言:javascript
复制
>>> print(wp['ac'])
Traceback (most recent call last):
...
ValueError: Subnode name must be in ['a', 'd'], not 'c'.

产生一个value error。

获取子节点属性:

小波包树对象是一个树状结构,拥有一系列子节点对象。小波包只是节点类中的一个特殊子类。 小波包树的节点可以通过以下操作符来进行访问:obj[x] (Node.__ getitem__()).每一个节点都有一系列属性:data,path,node_name,parent,level,maxlevel 和 mode。

代码语言:javascript
复制
>>> x = [1, 2, 3, 4, 5, 6, 7, 8]
>>> wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric')
代码语言:javascript
复制
>>> print(wp['ad'].data)
[-2. -2.]
代码语言:javascript
复制
>>> print(wp['ad'].path)
ad
代码语言:javascript
复制
>>> print(wp['ad'].node_name)
d
代码语言:javascript
复制
>>> print(wp['ad'].parent.path)
a
代码语言:javascript
复制
>>> print(wp['ad'].level)
2
代码语言:javascript
复制
>>> print(wp['ad'].maxlevel)
3
代码语言:javascript
复制
>>> print(wp['ad'].mode)
symmetric

提取符合条件的节点:

代码语言:javascript
复制
>>> x = [1, 2, 3, 4, 5, 6, 7, 8]
>>> wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric')

我们可以按自然顺序得到特定层上的所有节点:

代码语言:javascript
复制
>>> print([node.path for node in wp.get_level(3, 'natural')])
['aaa', 'aad', 'ada', 'add', 'daa', 'dad', 'dda', 'ddd']

或者按频带频率进行排序:

代码语言:javascript
复制
>>> print([node.path for node in wp.get_level(3, 'freq')])
['aaa', 'aad', 'add', 'ada', 'dda', 'ddd', 'dad', 'daa']

注意,WaveletPacket.get_level()也会执行自动分解,直到达到指定的级别。

从小波包树重构信号:

代码语言:javascript
复制
>>> x = [1, 2, 3, 4, 5, 6, 7, 8]
>>> wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric')

现在新建一个小波包树,并且给它的节点赋一些值。

代码语言:javascript
复制
>>> new_wp = pywt.WaveletPacket(data=None, wavelet='db1', mode='symmetric')

>>> new_wp['aa'] = wp['aa'].data
>>> new_wp['ad'] = [-2., -2.]

为了方便,也可以从节点对象中自动提取数据:

代码语言:javascript
复制
>>> new_wp['d'] = wp['d']

接下来,对aa,ad,d这三个节点包中的数据进行重构。

代码语言:javascript
复制
>>> print(new_wp.reconstruct(update=False))
[ 1.  2.  3.  4.  5.  6.  7.  8.]

如果reconstruct方法中的update参数被设置为False,那么根节点的数据将不会被更新。

代码语言:javascript
复制
>>> print(new_wp.data)
None

否则,根节点的data属性将被设置为重建后的数据。

代码语言:javascript
复制
>>> print(new_wp.reconstruct(update=True))
[ 1.  2.  3.  4.  5.  6.  7.  8.]
>>> print(new_wp.data)
[ 1.  2.  3.  4.  5.  6.  7.  8.]
代码语言:javascript
复制
>>> print([n.path for n in new_wp.get_leaf_nodes(False)])
['aa', 'ad', 'd']
代码语言:javascript
复制
>>> print([n.path for n in new_wp.get_leaf_nodes(True)])
['aaa', 'aad', 'ada', 'add', 'daa', 'dad', 'dda', 'ddd']

从小波包树中移除节点:

代码语言:javascript
复制
>>> x = [1, 2, 3, 4, 5, 6, 7, 8]
>>> wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric')

首先,从一个二层的小波包树分解开始,树中的子节点是:

代码语言:javascript
复制
>>> dummy = wp.get_level(2)
>>> for n in wp.get_leaf_nodes(False):
...     print(n.path, format_array(n.data))
aa [  5.  13.]
ad [-2. -2.]
da [-1. -1.]
dd [ 0.  0.]
代码语言:javascript
复制
>>> node = wp['ad']
>>> print(node)
ad: [-2. -2.]

要从WP树中删除一个节点,可以使用Python的del obj[x] (node . __ delitem __):

代码语言:javascript
复制
>>> del wp['ad']

于是,树中剩余的节点为:

代码语言:javascript
复制
>>> for n in wp.get_leaf_nodes():
...     print(n.path, format_array(n.data))
aa [  5.  13.]
da [-1. -1.]
dd [ 0.  0.]

如果此时重构信号的话:

代码语言:javascript
复制
>>> print(wp.reconstruct())
[ 2.  3.  2.  3.  6.  7.  6.  7.]

现在恢复删除的节点及其对应的值:

代码语言:javascript
复制
>>> wp['ad'].data = node.data

打印叶子节点和重建后的信号,确认重构信号是正确的。

代码语言:javascript
复制
>>> for n in wp.get_leaf_nodes(False):
...     print(n.path, format_array(n.data))
aa [  5.  13.]
ad [-2. -2.]
da [-1. -1.]
dd [ 0.  0.]
代码语言:javascript
复制
>>> print(wp.reconstruct())
[ 1.  2.  3.  4.  5.  6.  7.  8.]

惰性求值 我的理解是: Lazy evaluation(惰性求值)的意思是把一个表达式本身存储起来,并不进行求值。需要求值的时候再明确的让它求值。 如:(defparameter temp (+ 2 3))这种情况下,temp等于5 如果用Lazy, (defparameter temp (lazy (+ 2 3)) 这时temp等于一个表达式,再需要对它进行求值的时候用(force temp)把它evaluation。 它显而易见的好处是表达式只需要在需要的时候求值,可以避免了重复的计算。这个说法也不太准确,准确点的好处是在使用递归方法求值时避免了提前进行没有用到的求值。 注意:本节仅用于演示pywt的内部组件。不要依懒于本例中所示的对节点的属性访问。

代码语言:javascript
复制
>>> x = [1, 2, 3, 4, 5, 6, 7, 8]
>>> wp = pywt.WaveletPacket(data=x, wavelet='db1', mode='symmetric')
  • 1 一开始wp的属性a是None。
代码语言:javascript
复制
>>> print(wp.a)
None

要记得千万不要依懒于这种属性访问方式。

  • 2 .首先尝试访问节点,它是通过分解其父节点(wp对象本身)计算出来的。
代码语言:javascript
复制
>>> print(wp['a'])
a: [  2.12132034   4.94974747   7.77817459  10.60660172]
  • 3 .现在wp的属性a已经被设置成了新创建的节点。
代码语言:javascript
复制
>>> print(wp.a)
a: [  2.12132034   4.94974747   7.77817459  10.60660172]

节点d也同样被设置成了新创建的节点。

代码语言:javascript
复制
>>> print(wp.d)
d: [-0.70710678 -0.70710678 -0.70710678 -0.70710678]

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/209951.html原文链接:https://javaforall.cn

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022年10月25日,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档