Python列表元素替换为其余元素乘积的函数问题及非numpy解法咨询

错误原因 #

你用numpy代码出错的核心问题是直接在原始数组上修改元素，导致后续计算其他位置的乘积时，调用的是已经被修改过的数值，而非输入的原始元素。比如第一个元素被替换为其他元素的乘积后，第二个元素的计算会用到这个已修改的第一个元素，而非原始值，最终结果完全偏离预期。

举个实际例子：输入[1,2,3,4]，若直接修改原数组，第一次把第一个元素改成2*3*4=24，此时数组变成[24,2,3,4]，后续计算第二个元素时，会用24*3*4=288，这显然和预期的1*3*4=12不符。

不使用numpy的解决方案 #

方法1：总乘积除法（需处理含0场景） #

先计算所有元素的总乘积，再用总乘积除以每个元素得到结果。但要注意输入含0的特殊情况：有多个0时结果全为0；有一个0时，仅对应0的位置结果为其他元素乘积，其余为0。

代码示例：

def product_except_self(nums):
    total_product = 1
    zero_count = 0
    zero_index = -1
    for idx, num in enumerate(nums):
        if num == 0:
            zero_count += 1
            zero_index = idx
            continue
        total_product *= num
    
    if zero_count > 1:
        return [0] * len(nums)
    elif zero_count == 1:
        result = [0] * len(nums)
        result[zero_index] = total_product
        return result
    else:
        return [total_product // num for num in nums]

方法2：左右乘积遍历（无除法，适配所有场景） #

通过两次遍历分别计算每个元素的左侧乘积、右侧乘积，最后将对应位置的左右乘积相乘得到结果。这种方法无需除法，不受0的影响，时间复杂度为O(n)。

代码示例：

def product_except_self(nums):
    n = len(nums)
    left_products = [1] * n
    right_products = [1] * n
    
    # 计算左侧元素乘积
    for i in range(1, n):
        left_products[i] = left_products[i-1] * nums[i-1]
    
    # 计算右侧元素乘积
    for i in range(n-2, -1, -1):
        right_products[i] = right_products[i+1] * nums[i+1]
    
    # 左右乘积合并得到结果
    return [left_products[i] * right_products[i] for i in range(n)]

示例验证 #

输入：[1,2,3,4]
期望输出：[24,12,8,6]

用方法2运行后：

• left_products: [1,1,2,6]
• right_products: [24,12,4,1]
• 对应位置相乘后结果与期望完全一致。

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