zip(*...)
是“转置”列表(或元组)列表的便捷方法。与 A[x,y]
相同A[(x,y)]
。
In [397]: lst = [(0,1), (0, 2), (1, 0), (1, 3), (2,1)]
In [398]: tuple(zip(*lst)) # make a tuple of tuples (or lists)
Out[398]: ((0, 0, 1, 1, 2), (1, 2, 0, 3, 1))
In [399]: A=np.zeros((3,4),dtype=bool) # make an array of False
In [400]: A[tuple(zip(*lst))] = True # assign True to the 5 values
In [401]: A
Out[401]:
array([[False, True, True, False],
[ True, False, False, True],
[False, True, False, False]], dtype=bool)