信息论与编码

已知信源概率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