已知信源概率P(xi)和条件概率(也称转移概率)P(yj|xi),请编写一个程序,能比较快捷地计算出联合概率P(xi,yj)以及后验概率P(xi|yj)。
clc,clear
PX = input("请输入信源概率的列向量P(xi):");
PY_X = input("请输入纵为Y横为X的条件概率(也称转移概率)P(yj|xi):");
PXY = PX .* PY_X;%求P(X,Y)的联合概率
[m,n] = size(PXY);%这行可以省略,因为这些代码都是用矩阵运算,如果想尝试用循环可以使用这个语句得到矩阵的行和列的大小
S = sum(PXY,2);%对P(X,Y)进行行求和,sum(A,2)代表行求和,sum(A,1)代表列求和,A为矩阵
PX_Y = PXY ./ S;%求P(X/Y)的条件概率
PY = PXY ./ PX_Y;%求P(Y)的概率
%可以不用写这个代码块,题目没有说明
HX = - PX .* log2(PX);%求H(X)的信息熵
HY_X = - PXY .* log2(PY_X);%求H(Y/X)的信息熵
HXY = HX + HY_X;%求H(X,Y)的信息熵
HY = - PY .* log2(PY);%求H(Y)的信息熵
HX_Y = - PXY .* log2(PX_Y);%求H(X/Y)的信息熵
disp("联合概率P(xi,yj)为:") %打印输出答案
PXY
disp("后验概率P(xi|yj)为:")
PX_Y