如何使用二进制列表拆分numpy数组？

Question

如果我有一个numpy数组，请说

A = [[1,2],[3,4],[5,6],[7,8]]

和一个列表

L = [1,0,1,1]

是否有一种方法可以将A下axis0拆分为L中的1/0？这将是我想要的结果：

A1 = [[1,2],[5,6],[7,8]]
A2 = [[3,4]]

Answer 1

由于L是二进制，因此您可以将L转换为布尔类型，然后使用boolean indexing：

A = np.array([[1,2],[3,4],[5,6],[7,8]])
L = np.array([1,0,1,1])

L = L.astype(bool)
A1, A2 = A[L], A[~L]

A1
#array([[1, 2],
#       [5, 6],
#       [7, 8]])

A2
#array([[3, 4]])