列表理解中的python双迭代器

如何解决列表理解中的python双迭代器

所以如果我在 S 的所有索引位置都有 1，其余的都是 0。新字符串的长度将为 N。我如何在 Python 列表理解中做到这一点

S= [2,5,8]
N= 10
Desired output: [0,1,0]

我的代码：

newlist = [1 if x.__index__() == S[i]  else 0 for i in range(len(S)) for x in range(N)]

My Output: [0,0]

解决方法

直接的解决方案是不使用列表推导式：

result = [0] * N
for i in S:
    result[i - 1] = 1

请注意，需要 i - 1 将列表 S 中从 1 开始的索引转换为从 0 开始的索引。如果您的列表中有基于 0 的索引（这对于大多数用途来说会更简单），那么您可以只写 result[i] = 1。

,

output = [1 if i+1 in S else 0 for i in range(N)]

,

如果您的 S 列表使用基于 1 的索引

[int(i in S) for i in range(1,N+1)]