关于Numpy数组动态形状下通用元素相乘解决方案的技术问询

现有逻辑表定义 #

import numpy as np
table1 = np.zeros((4,3)); table2 = np.zeros((4,3)); table3 = np.zeros((4,3));
table1[0] = 1; table2[1] = 1; table3[1] = table3[3] = 1;

动态变化的equations数组 #

模拟过程中存在一组形状动态变化的数组（称为equations），其形状不超过逻辑表的维度（例如(2,3)、(2,2)等）：

eq1 =[[a,b,c], [d,e,f]]; eq2 = [[g,h,i],[j,k,l]]; eq3 = [[m,n,o],[p,q,r]]; 
# 结果数组初始化
result = np.zeros((4,3))

期望目标 #

通过逻辑表将不同equations分配到对应行并叠加，最终得到目标结果：

result = [[a,b,c], [g +m,h+n,i+o], [0,0,0], [p,q,r]];

核心需求 #

原代码因数组形状不匹配无法运行，需实现通用方案：在模拟循环中，当逻辑表（如shape=(4,3)）与形状更小的eq数组相乘时，逻辑表中超出eq范围的元素保持原样，实现如下循环逻辑：

for time_simulation:
   Update(eq1,eq2,eq3)
   result = table1*eq1 + table2*eq2 + table3*eq3
   Use(result)

解决方案 #

方案1：自动填充对齐（适用于eq形状为逻辑表前缀的场景） #

先将eq数组填充至与逻辑表相同的形状，填充值为0，这样与逻辑表相乘时不会改变原逻辑表的0元素，再进行叠加计算。实现一个辅助函数完成填充：

def pad_to_match(eq, target_shape):
    # 创建目标形状的0数组
    padded_eq = np.zeros(target_shape, dtype=np.array(eq).dtype)
    # 计算eq与目标形状的重叠切片范围
    slice_dims = tuple(slice(0, min(eq_dim, tgt_dim)) for eq_dim, tgt_dim in zip(np.array(eq).shape, target_shape))
    padded_eq[slice_dims] = eq
    return padded_eq

修改循环逻辑：

# 假设time_simulation是可迭代对象
for _ in time_simulation:
    Update(eq1, eq2, eq3)
    # 将每个eq填充至result的形状
    padded_eq1 = pad_to_match(eq1, result.shape)
    padded_eq2 = pad_to_match(eq2, result.shape)
    padded_eq3 = pad_to_match(eq3, result.shape)
    # 执行元素相乘与叠加
    result = table1 * padded_eq1 + table2 * padded_eq2 + table3 * padded_eq3
    Use(result)

方案2：索引赋值叠加（适用于eq形状不规则的场景） #

直接通过逻辑表的非零位置索引，将eq数组的元素对应赋值/叠加到result中，无需考虑eq的形状，只需保证eq的有效元素数量与逻辑表的非零元素数量一致：

for _ in time_simulation:
    Update(eq1, eq2, eq3)
    # 初始化结果数组
    result = np.zeros((4,3))
    # 处理table1：非零位置赋值eq1的元素
    idx1 = table1 != 0
    result[idx1] = np.array(eq1).ravel()[:np.sum(idx1)]
    # 处理table2：非零位置叠加eq2的元素
    idx2 = table2 != 0
    result[idx2] += np.array(eq2).ravel()[:np.sum(idx2)]
    # 处理table3：非零位置叠加eq3的元素
    idx3 = table3 != 0
    result[idx3] += np.array(eq3).ravel()[:np.sum(idx3)]
    Use(result)

内容的提问来源于stack exchange，提问作者zizou