在Numpy中使用np.logical_and進行子選擇

Question

我繪制了（x，y）圖表，並希望當從A到B的x的y大於C時返回“ True”（“區間中存在x使得y（x）> C）。下面的代碼不起作用。 我該怎么做？

d_for_hours=density[np.logical_and(y>C,x>=A, x <=B)].all()

示例輸出：

對於C = 0.02和A = 9且B = 13，輸出應為True

對於C = 0.05且A = 9且B = 13，輸出應為False

Answer 1

您想使用any而不是all來檢查y值是否大於C 在此之前，您需要將y限制為與條件（在A和B之間）相匹配的x的索引：

# create data
x = np.array(range(20))
y = np.array(19 * [0] + [1])

(y[np.logical_and(x>=9, x<=13)] >= 0.05).any()  # False
(y[np.logical_and(x>=9, x<=20)] >= 0.05).any()  # True