根据基础 Python 中的元组列表使用理解列表分配值

问题描述

我有这个零矩阵和一个元组列表：

my_array = [[0 for col in range(5)] for row in range(10)]
locations = [(2,1),(6,3)]

我想根据基本 python 中每个元组的值（它们是索引）将零转换为 1。我试过这个：

my_array = [[i for i in enumerate(locations)]j for j in enumerate(locations) my_array[i][j]=1]

但我收到此错误：

my_array = [[i for i in enumerate(locations)]j for j in enumerate(locations) my_array[i][j]=1]
                                                 ^
SyntaxError: invalid Syntax

我试图理解我做错了什么，拜托，你能帮我吗？提前致谢。

解决方法

除非您要更改大量元素，否则只需使用常规循环即可。

for i,j in locations:
    my_array[i][j] = 1

如果您要重新创建整个列表，则无论如何都必须遍历所有索引，而不仅仅是 locations 中的索引，因此您不妨执行类似

的操作>

loc_set = set(locations)
my_array = [[1 if (i,j) in set(locations) else 0 for j in range(5)] for i in range(5)]