python - 从 Python 中的二维数组中删除行

Question

我有一个 585L、2L 的 Python numpy 数组。数据的组织方式与此示例类似。

0,0
1,0
2,1
3,0
4,0
5,1
...

我想创建一个数组，删除秒列中存在 0 的所有行。因此，最终假装的结果是:

2,1
5,1

我已经阅读了一些相关示例，但我仍在为此苦苦挣扎。

Answer 1

既然你提到你的结构是一个 numpy 数组，而不是一个列表，我会使用 numpy 逻辑索引来只选择你关心的值。

>>> import numpy as np
>>> x = [[0,0], [1,0], [2,1], [3,0], [4,0], [5,1]] # Create dummy list
>>> x = np.asarray(x) # Convert list to numpy array
>>> x[x[:, 1] != 0] # Select out items whose second column don't contain zeroes

 array([[2, 1],
   [5, 1]])