如何截断大于指定值的numpy数组？

Question

import numpy as np
a=np.linspace(1,10,num=10)
truncatevalue = 5.5

a_truncated=([1., 2., 3., 4., 5.])

import numpy as np
a=np.linspace(1,10,num=10)
b=np.array([19, 17, 15, 14, 29, 33, 28, 4, 90, 6])
truncatevalue = 5.5

a_truncated=([1., 2., 3., 4., 5.])
b_truncated=([19, 17, 15, 14, 29])

我不知道它是否会像重复需要做什么来获得_truncated那么简单，所以我也想包含它，以防需要做一些不同的事情。

Answer 1

可以使用布尔索引：

>>> a = np.linspace(1, 10, num=10)
>>> truncatevalue = 5.5
>>> a_truncated = a[a < truncatevalue]
>>> a_truncated
array([ 1.,  2.,  3.,  4.,  5.])

本质上，a 返回一个布尔数组，指示 a的元素是否满足条件。使用此布尔数组索引 a将返回 a的视图，其中每个元素的索引为 true。

所以对于问题的第二部分，你需要做的就是：

>>> b = np.array([19, 17, 15, 14, 29, 33, 28, 4, 90, 6])
>>> b_truncated = b[a < truncatevalue]
>>> b_truncated
array([19, 17, 15, 14, 29])