实现两个二维数组之间的加权平均

import numpy as np

def weighted_average(arr1, arr2, weights):
    # 确保维度相同
    assert arr1.shape == arr2.shape == weights.shape
    
    # 计算加权和
    weighted_sum = np.sum(arr1 * weights) + np.sum(arr2 * weights)
    
    # 计算总权重
    total_weight = np.sum(weights)
    
    # 计算加权平均值
    weighted_avg = weighted_sum / total_weight
    
    return weighted_avg

# 示例数据
arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([[5, 6], [7, 8]])
weights = np.array([[0.5, 0.5], [0.25, 0.75]])

# 计算加权平均值
result = weighted_average(arr1, arr2, weights)
print("加权平均值：", result)