python - python中大于x且小于y的元素的表达式，全部在一个return语句中。没有循环。Python

Question

返回数组A中大于 x 且小于 y 的项目数。A是一维浮点数组，x 和 y 是浮点数。例如F6(np.array([1.1, 2.2, 3.3, 2.3]), 2, 3)应该返回2。

def F(A,x,y):
     return ________?

只能使用这条线。在蟒蛇

score 5 · Accepted Answer

这是一种将其作为矢量化操作的方法：

import numpy as np

def F(A, x, y):
    return np.sum((A > x) & (A < y))

每个比较都返回一个布尔数组。表达式按元素对这两个数组进行 AND 运算并将结果相加。

score 4 · Accepted Answer

我对numpy一无所知，但是在香草python中：

def F(A, x, y):
    return sum(x < v < y for v in A)

True这使用了这样一个事实，即您可以通过求和来计算布尔数组中 s 的数量，因为int(True) == 1和int(False) == 0。

score 2 · Accepted Answer

2

列表理解是一种选择。

def F(A,x,y):
    return len([i for i in A if i > x and i < y])

于 2012-10-31T23:39:52.727 回答

3 回答 3