您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

从元组生成二维布尔数组

从元组生成二维布尔数组

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)
其他 2022/1/1 18:45:17 有447人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶