用Python，计算加权熵Hw（X）

Wyecho

于 2023-04-06 16:50:38 发布

阅读量233

点赞数 1

CC 4.0 BY-SA版权

分类专栏： python-信息论文章标签： python numpy

本文链接：https://blog.csdn.net/Wyecho/article/details/129993250

本文介绍了如何使用Python和NumPy库计算加权熵。通过给出的代码示例，展示了如何处理概率分布和权重，得到加权熵Hw(X)的计算结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

输入随机变量X的概率分布和权重，输出 $H_{w}\left ( X \right ) = -\sum_{i=1}^{q} w_{i}p_{i}logP_{i}$ ，其中 $P_{i}$ 是事件 $a_{i}$ 的概率， $w_{i}$ 是事件 $a_{i}$ 的权重。

import numpy as np
 
p = eval(input("请输入事件的p(X)：[p1,p2,...]"))
w = eval(input("请输入事件的p(X)的权重w：[w1,w2,...]"))
SUM = 0
 
if sum(p) != 1 or len(p)!= len(w): #要求概率分布的和=1，且p和权重的数量相等
    print("概率分布错误")
for i in range(0,len(p)):
    if p[i] > 1 or p[i] <= 0: #概率在（0，1]
        print("概率分布错误")
    elif w[i] < 0:
        p