主页 / 问题 / 1427851
Accepted
фан-клуб Берлина
Asked:2022-09-08 19:40:01 +0000 UTC2022-09-08 19:40:01 +0000 UTC

通过 numpy 条件删除列

  • 772

大家好，我想通过条件删除nxm矩阵中的一列，但是实现有问题。我认为它会以这种方式工作，但解释器会抛出一个错误：

a = a[a[:,[0,m-1]]%5 == 0]

你能建议如何解决这个问题吗？根据任务，需要找到一个是5的倍数的元素，并用它删除一列

python numpy

1 个回答

  1. Best Answer

    删除所有具有 5 倍数的列：

    import numpy as np
arr=np.arange(24).reshape(3,8)

    我们有一个数组：

    [[ 0  1  2  3  4  5  6  7]
 [ 8  9 10 11 12 13 14 15]
 [16 17 18 19 20 21 22 23]]

    根据给定条件删除：

    np.delete(arr,np.where(arr%5 == 0)[1],1)

    结果：

    [[ 1  3  6]
 [ 9 11 14]
 [17 19 22]]

    事实上，这在任何谈论 numpy 的文章（更不用说一本书）中都有描述。我建议从这个开始。

    • 2

相关问题