我有一个清单(aList)
我需要创建另一个长度与aList(checkColumn)相同的列表,并在索引列表(indexList)中定义的位置添加1,未在indexList中定义的位置应为0
输入:
aList = [70, 2, 4, 45, 7 , 55, 61, 45]
indexList = [0, 5, 1]
所需输出:
checkColumn = [1,1,0,0,0,1]
我一直在尝试以下代码,但输出为[1,1]
for a in len(aList):
if a == indexList[a]:
checkColumn = checkColumn[:a] + [1] + checkColumn[a:]
else:
checkColumn = checkColumn[:a] + [0] + checkColumn[a:]
我已经尝试过使用checkColumn.insert(a,1)并得到相同的结果。
谢谢您的帮助!
最佳答案
这会有所帮助吗?
首先,您将checkColumn初始化为0
checkColumn = [0] * len(aList)
然后,遍历indexList并更新checkColumn
for idx in indexList:
checkColumn[idx] = 1
干杯!