Python：基于Cb_max1非零元素修正sigma嵌套列表的问题

需求：现有两个嵌套列表sigma和Cb_max1，需要将sigma中与Cb_max1每个非零元素对应的位置设置为0.021。

原代码 #

sigma = [[0, 0, 0, 0, 0, 0.005, 0]]
Cb_max1 = [[0.0009847628716281426, 0.0009847628716281426, 0, 0, 0, 0.0009800221856714917, 0]]
sigma = [i if j != 0 else 0.021 for i, j in zip(Cb_max1, sigma[0])]
print(sigma)

当前输出 #

[0.021]

预期输出 #

[[0.021, 0.021, 0, 0, 0, 0.005, 0]]

错误原因 #

原代码中zip(Cb_max1, sigma[0])的配对逻辑错误：Cb_max1是嵌套列表，直接用它参与zip时，只会取到它内部的唯一子列表作为第一个元素，和sigma[0]的第一个元素配对，循环仅执行一次，导致输出结果不符合预期。

修正代码 #

sigma = [[0, 0, 0, 0, 0, 0.005, 0]]
Cb_max1 = [[0.0009847628716281426, 0.0009847628716281426, 0, 0, 0, 0.0009800221856714917, 0]]
# 取两个列表的子列表进行配对，生成新的子列表后再嵌套
sigma = [[0.021 if cb != 0 else s for s, cb in zip(sigma[0], Cb_max1[0])]]
print(sigma)

代码说明 #

1. 用sigma[0]和Cb_max1[0]取出两个一维子列表，通过zip逐个配对元素；
2. 遍历配对后的元素对：如果Cb_max1中的元素cb非零，就将sigma对应位置设为0.021，否则保留原sigma的元素s；
3. 将生成的新一维列表重新嵌套成二维列表，赋值给sigma，得到预期格式的输出。

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