主页 / 问题 / 822599
Accepted
firet
Asked:2020-05-03 18:32:48 +0000 UTC2020-05-03 18:32:48 +0000 UTC

按第 N 列元素过滤二维数组

  • 772

数组从文件中加载，格式为：[[1,2,3,4], [5,6,7,8]..].

如何过滤它，例如，每个内部数组的第三个元素？

python

1 个回答

  1. Best Answer

    例子：

    In [199]: a = np.arange(40).reshape(-1, 4)

In [200]: a
Out[200]:
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15],
       [16, 17, 18, 19],
       [20, 21, 22, 23],
       [24, 25, 26, 27],
       [28, 29, 30, 31],
       [32, 33, 34, 35],
       [36, 37, 38, 39]])

    2我们只留下第三列（带有索引）的元素可被无余数整除的那些行3：

    In [201]: a[a[:, 2] % 3 == 0]
Out[201]:
array([[ 4,  5,  6,  7],
       [16, 17, 18, 19],
       [28, 29, 30, 31]])
    • 2

相关问题