编写一个通用的用于计算平均互信息量的程序 输入：以变量描述的信源概率分布和信道转移概率；输出： 给定变量 i 、 j 值下的交互信息量；平均互信息量

好的，下面是一个通用的用于计算平均互信息量的程序：

import math

def mutual_information(p, q):
    mi = 0
    for i in range(len(p)):
        for j in range(len(q)):
            if p[i][j] != 0 and q[i][j] != 0:
                mi += p[i][j] * math.log2(p[i][j] / q[i][j])
    return mi

def average_mutual_information(p, q):
    ami = 0
    for i in range(len(p)):
        for j in range(len(q)):
            if p[i][j] != 0:
                ami += p[i][j] * mutual_information(p, q)
    return ami

其中，p 和 q 分别表示信源概率分布和信道转移概率，均为二维数组。函数 mutual_information 用于计算给定变量 i、j 值下的交互信息量，函数 average_mutual_information 则用于计算平均互信息量。

注意，这里假设了 p 和 q 的行数和列数相同。如果不同，需要对代码进行相应的修改。