如何將元組列表轉換為掩碼？

Question

• 想象一下，我有一個形狀為numpy的數組（720，1280）：

  grid = np.zeros((720, 1280))

• 然后我有一個元組列表，看起來像這樣：

  x_y_pairs_to_activate = [(241, 623), (390, 143), (313, 406)]

如何將該元組列表轉換為掩碼，以便激活各自的坐標？

grid[x_y_pairs_to_activate] = 1 # something like that

Answer 1

您需要轉置坐標列表，以便在一個列表中擁有所有第一坐標，而在另一個列表中擁有所有第二坐標。

x_coords = [c[0] for c in x_y_pairs_to_activate]
y_coords = [c[1] for c in x_y_pairs_to_activate]
grid[x_coords, y_coords] = 1

或者使用更多的“ numpythonic”語言

mask = np.array(x_y_pairs_to_activate).transpose()
grid[mask] = 1