问题描述
所以如果我在 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)]