Python:按列值分隔矩阵

Question

我有一个包含 3 列的矩阵 A，看起来有点像，但更大:

[[10 15 1.0]
 [21 13 1.0]
 [9  14 0.0]
 [14 24 1.0]
 [21 31 0.0]
 ...]

我想创建两个单独的矩阵:一个包含第三列 = 0.0 的所有数据，另一个包含第三列 = 1.0 的所有数据。因此，本质上是按第三列中的值 0.0 或 1.0 拆分数据。

Answer 1

如果您使用 Numpy ，首先找到第三列具有所需值的行，然后使用 indexing 提取行.

演示

>>> import numpy
>>> A = numpy.array([[1, 0, 1],
                     [2, 0, 1],
                     [3, 0, 0],
                     [4, 0, 0],
                     [5, 0, 0]])
>>> A1 = A[A[:, 2] == 1, :] # extract all rows with the third column 1
>>> A0 = A[A[:, 2] == 0, :] # extract all rows with the third column 0
>>> A0
array([[3, 0, 0],
       [4, 0, 0],
       [5, 0, 0]])
>>> A1
array([[1, 0, 1],
       [2, 0, 1]])