python - 当我们使用列表理解时如何保持原始列表的形状？

Question

我有一个形状如下的零列表:

yp = numpy.zeros(5, 2, 2), dtype = complex)

我使用两个 for 循环修改了它，如下所示:

for a in range(0,5):
    for b in range(0, 2):
        yp[a, b, b] = numpy.sum(F[a, b,:])

我怎样才能使用列表理解来做同样的事情？显然，使用下面的行会改变我原来列表的形状。

yp = [numpy.sum(F[a, b,:]) for a in range(0,5) for b in range(0, 2)]   

Answer 1

您的原始循环无法转换为列表理解，因为它没有分配 numpy 矩阵的所有元素。列表推导总是会生成完整的列表(或列表的列表)。

如果您的目标是用零填充未赋值的元素，那么相应的列表理解可能是这样的:

[ [ [0,0,numpy.sum(y_network[a, b, :])] for b in range(2)] for a in range(5)]

不同之处在于，每个理解级别都会生成一个列表，该列表本身使用对其元素的理解。

请注意，这并不完全相同，因为 numpy.sum(y_network[a, b, :]) 的值始终放在第 3 维的最后一个元素中在索引 b 处。您可以使用更多代码来左右填充零，但这会使它变得复杂且不可读，这几乎会破坏使用列表理解的目的。

你也可以为赋值写一个循环:

for b in range(2): yl[:,b,b] = np.sum(network[:,b,:],1)