是否有更Pythonic的写法替代双重循环打印NumPy数组元素配对？

你需要实现的是两个NumPy数组元素的笛卡尔积逐对打印效果，完全可以移除手写的两层嵌套循环，以下两种实现的输出结果和原代码完全一致：

标准库通用方案（无额外NumPy操作） #

• 直接借助Python标准库itertools.product生成所有元素的配对组合，仅用一层循环即可完成遍历，适配所有可迭代类型，不依赖NumPy的特殊API

import itertools
import numpy as np

sigma_n = np.array([0.01, 0.1])
ls = np.array([0.1, 0.2])

for sigma_ni, ls_i in itertools.product(sigma_n, ls):
    print(f"{sigma_ni},{ls_i}")

注意：itertools.product的参数顺序需要和原嵌套循环的外层、内层数组顺序保持一致，否则输出顺序会和原代码存在差异。

NumPy生态原生方案 #

• 针对本身就在NumPy数据处理流程的场景，可以借助NumPy的网格广播API一次性生成所有配对，甚至可以完全移除Python层面的循环，做批量向量化输出

import numpy as np

sigma_n = np.array([0.01, 0.1])
ls = np.array([0.1, 0.2])

# 生成和原嵌套循环遍历顺序完全一致的配对数组
pair_grid = np.stack(
    np.meshgrid(sigma_n, ls, indexing="ij"), 
    axis=-1
).reshape(-1, 2)

# 单层循环遍历打印
for sigma_ni, ls_i in pair_grid:
    print(f"{sigma_ni},{ls_i}")

# 极致向量化版本：完全移除Python循环，一次性拼接所有输出字符串
print_str = "\n".join(
    np.char.add(np.char.add(pair_grid[:,0].astype(str), ","), pair_grid[:,1].astype(str))
)
print(print_str)

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