帮我写一个python语言用公式计算布尔函数Walsh谱的算法

在Python中，我们可以使用numpy库来计算布尔函数的Walsh谱。Walsh谱是一种将二进制函数映射到复数域的技术，在密码学、编码理论和信号处理等领域有应用。以下是一个简单的算法示例：

import numpy as np

def walsh_matrix(n):
    """
    创建n阶的沃尔什矩阵
    """
    w = np.zeros((2**n, 2**n), dtype=complex)
    for i in range(2**n):
        for j in range(2**n):
            # Walsh系数计算
            w[i][j] = (-1) ** (i * j % (2**n))
    return w

def walsh_spectrum(funct, n):
    """
    计算布尔函数f(x_1, x_2, ..., x_n)的沃尔什谱
    funct: 输入一个接受n位输入的布尔函数，返回一个整数
    n: 输入变量的位数
    """
    walsh_matrix_n = walsh_matrix(n)
    spectrum = np.abs(np.fft.fft(walsh_matrix_n)[funct]) ** 2
    return spectrum

# 示例使用：
def my_bool_function(binary_input):
    # 这里替换为实际的布尔函数
    return binary_input[0] ^ binary_input[1]

n_bits = 2  # 假设我们有两个输入比特
binary_inputs = np.array([0, 1], dtype=int)
spectral_coefficients = walsh_spectrum(my_bool_function, n_bits)

print("沃尔什谱:", spectral_coefficients)