python - 从其他列表创建列表的更简单方法

Question

我有 3 个列表:x、y 和 z。所有的大小都相同，但我需要创建新的 x、y 和 z 列表，具体取决于 z 的每个索引处的值，如下所示:

xNew = []
yNew = []
zNew = []
i = 0
for value in z:
    if value > 0:
        xNew.append(x[i])
        yNew.append(y[i])
        zNew.append(z[i])
    i += 1

有谁知道是否有更简洁、甚至更高效的方法来执行上述计算？

Answer 1

可能是最直接的方式:

it = zip(x, y, z)
xnew, ynew, znew = zip(*(t for t in it if t[-1] > 0))

我们使用 the zip function两次恢复数据的原始结构。

---

zip(x, y, z) 创建一个新的迭代器对象，它产生三元组。

(t for t in it if t[-1] > 0) 过滤三元组(t[-1] 是值 ).

zip(*(...)) 正好产生 3 个元组和 xnew、ynew 和 znew接收它们。