Python中求列表元素两两乘积（含自身）的方法咨询

嘿，刚好碰到过类似的需求！首先明确说：Python标准库中没有直接实现这个需求的内置函数，不过自己写个自定义函数非常简单，而且针对你8000条数据的场景，我们可以兼顾简洁性和效率。

纯Python实现方案 #

最直观的方式是遍历每个元素，然后和它自己以及后面的所有元素相乘，把结果收集起来。这里给两种写法：

循环+append（可读性好）

def pairwise_product_with_self(lst):
    result = []
    for idx, current_num in enumerate(lst):
        # 从当前位置开始，取后面所有元素（包括自己）
        for num in lst[idx:]:
            result.append(current_num * num)
    return result

# 测试你的示例
A = [1,2,3,4]
print(pairwise_product_with_self(A))
# 输出: [1, 2, 3, 4, 4, 6, 8, 9, 12, 16]

列表推导式（更简洁）

如果你喜欢一行式的写法，也可以用嵌套列表推导：

def pairwise_product_with_self(lst):
    return [lst[i] * lst[j] for i in range(len(lst)) for j in range(i, len(lst))]

针对8000条数据的效率优化 #

上面的纯Python方案逻辑简单，但对于8000条数据来说，计算量大概是(8000×8001)/2 ≈3200万次运算，虽然能跑，但如果想更快完成，可以用numpy的向量化操作——它比纯Python循环快得多：

import numpy as np

def pairwise_product_numpy(arr):
    np_arr = np.array(arr)
    # 用outer生成所有两两乘积的矩阵，再用triu提取上三角（含对角线）部分
    product_matrix = np.outer(np_arr, np_arr)
    upper_triangle = np.triu(product_matrix)
    # 把非零元素（就是我们要的i≤j的乘积）转成列表
    return upper_triangle[upper_triangle != 0].tolist()

# 测试示例
A = [1,2,3,4]
print(pairwise_product_numpy(A))
# 输出: [1, 2, 3, 4, 4, 6, 8, 9, 12, 16]

小建议 #

• 如果你的数据都是数值类型，优先选numpy方案，能节省不少时间；
• 纯Python方案不需要额外安装依赖，适合环境受限的场景；
• 两种方案都能准确生成你需要的i≤j组合的乘积，包括元素自身相乘的情况。

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