python计算香浓熵(直方图法)

import numpy as np
import matplotlib.pyplot as plt
import math

data = [160,162,183,174,172,161,171]
bins_size = 5  # 每个bin的宽度,可以自行设置
# 获取所有bin,np.max(data)+bins_size确保所有数值都有bin装,如果np.max(data),这里你会发现样本183没有bin可以装
num_bins = np.arange(np.min(data), np.max(data)+bins_size, bins_size)

# 获取样本频率:Xn
Xn, bins, patches = plt.hist(data, 
                             num_bins, 
                             density = False,  # 那么纵坐标表示落入这个bins的样本个数
                             # density = True, # 那么纵坐标表示落入这个bins的样本概率(密度)
                             facecolor='blue',  
                             alpha=0.5)
# 获取样本概率:Px
Px = Xn/np.size(data) 

# 计算香浓熵
shannonEnt = 0
for i in range(np.size(Px)):
    if (Px[i] != 0):  
        shannonEnt -= Px[i] * math.log(Px[i], 2)
print("shannonEnt:", shannonEnt)

香浓熵的原理看我另一篇博客:从熵到相位传递熵,附matlba和python代码

你可能感兴趣的:(python,开发语言,概率论,信息熵)