python numpy数组，满足条件的子数组问题

Question

例如，我有一个数组X = np.array([1,-3,5,0,9,12])

我想做一个像这样的函数。

def bigfunction(X)
    if X<0:
        return 99
    if X=>0 and X<=10
        return 100
    if X>10
        return 101

它还返回一个数组。在本例中[100,99,100,100,100,101]显然，这段代码是行不通的。非常重要的是我不能循环执行它。我想知道是否已经在 numpy 中实现了代码来解决这个问题。

Answer 1

您可以尝试np.select :

conds = [X < 0, X <= 10]

choices = [99, 100]

np.select(conds, choices, default=101)

这将返回:

array([100,  99, 100, 100, 100, 101])