丢弃数组中的某些值

Question

所以我有一个包含 20 个不同值的数组 (x)，我想去掉所有大于 10 的数字。我试过这个：

while x <= 10:
    x2 = x

但返回的是：“ValueError：具有多个元素的数组的真值不明确。使用 a.any() 或 a.all()”而且我不知道如何使用这些函数。关于摆脱数组中大于 10 的值的更好方法有什么建议吗？

Answer 1

您可以使用numpy 数组索引来过滤数组值。例如，

x = np.array([1, 2, 3, 4, 5])
x_2 = x[x < 3]

x_2 将返回的位置：

数组([1, 2])

您应该能够根据您的用例调整此示例。