时间戳

首页 > 解决方案 > 连接两个单维列表得到一个多维列表

python - 连接两个单维列表得到一个多维列表

问题描述

对于我的应用程序范围，我需要将两个一维数组连接成一个多维数组，两者都使用listPython 中的（最终嵌套）s 实现。连接必须打印第一个数组的元素与第二个数组的元素之间的所有可能组合。

vectA=[124,172,222,272,323,376,426,479,531]
vectB=[440,388,336,289,243,197,156,113,74]

预期结果是一个多维数组，其中包含（笛卡尔积）vectA的所有元素的组合。vectB

output=[[124,440],[124,388],[124,336],[124,289]...[172,440],[172,388]...]

标签： pythonpython-3.xlistnested-listscartesian-product

解决方案

使用itertools.product：

from itertools import product

vectA=[124,172,222,272,323,376,426,479,531]
vectB=[440,388,336,289,243,197,156,113,74]

output = list(product(vectA,vectB))
output = [list(i) for i in output]
print(output)

推荐阅读