如何在Python中将两个列表转换为数组坐标？

Question

在Python中，如何在给定2个轴列表的情况下构建坐标数组，以便输出数组包含所有可能的坐标对？

e.g。

ax1=[1,3,4]
ax2=[a,b]

"""
Code that combines them
"""

Combined_ax1 = [1,3,4,1,3,4,1,3,4]
Combined_ax2 = [a,a,a,b,b,b,c,c,c]

我需要这个，所以我可以在不使用多个for循环的情况下将combined_ax1和combined_ax2提供给函数。

Answer 1

此代码可以满足您的需求

import itertools

ax1=[1,3,4]
ax2=['a','b']

Combined_ax1, Combined_ax2 = zip(*itertools.product(ax1, ax2))

Answer 2

这可以通过使用列表推导来完成，如下所示：

cartesian_product = [(x, y) for x in ax1 for y in ax2]

此代码示例将返回包含所有可能坐标对的元组列表。