代码之家 › 专栏 › 技术社区 › blonc

使用Numpy argmax计数vs for循环

numpy python

blonc · 技术社区 · 4 年前

我目前使用类似的代码来确定比较

list_of_numbers = [29800.0, 29795.0, 29795.0, 29740.0, 29755.0, 29745.0]
high = 29980.0
lookback = 10
counter = 1

for number in list_of_numbers:
    if (high >= number) \
    and (counter < lookback):
        counter += 1
    else:
        break

结果 counter 幅度将是 7 。然而,这对大型数据阵列来说是非常费力的。所以,我一直在寻找解决方案,并想出了 np.argmax() ,但似乎存在问题。例如以下内容:

list_of_numbers = [29800.0, 29795.0, 29795.0, 29740.0, 29755.0, 29745.0]
np_list = np.array(list_of_numbers)
high = 29980.0

print(np.argmax(np_list > high) + 1)

这将获得输出 1 ,就像 argmax 假设。。但我想让它输出 7. 。是否有其他方法可以为if语句提供类似的输出?

1 回复 | 直到 4 年前

Ali_Sh 4 年前

你可以得到一个布尔数组 high >= number 使用NumPy:

list_of_numbers = [29800.0, 29795.0, 29795.0, 29740.0, 29755.0, 29745.0]
high = 29980.0
lookback = 10

boolean_arr = np.less_equal(np.array(list_of_numbers), high)

然后找到第一个在哪里错误论点中的满足 break 条件。此外,为了考虑对抗,您可以使用 np.cumsum 上 布尔数组 并找到满足指定条件的第一个自变量回环巨大因此,结果将是之间的较小值 break_arr 和 lookback_lim :

break_arr = np.where(boolean_arr == False)[0][0] + 1
lookback_lim = np.where(np.cumsum(boolean_arr) == lookback)[0][0] + 1
result = min(break_arr, lookback_lim)

如果你 list_of_numbers 没有比您指定的更大的值高的的限制 break_arr 或指定的回环超过中的值 np.cumsum(boolean_arr) 对于 lookback_lim ,上述代码将陷入如下错误,与有关 np.where :

IndexError:索引0超出大小为0的轴0的界限

可以由处理 try-except 或 if 语句,例如:

try:
    break_arr = np.where(boolean_arr == False)[0][0] + 1
except:
    break_arr = len(boolean_arr) + 1

try:
    lookback_lim = np.where(np.cumsum(boolean_arr) == lookback)[0][0] + 1
except:
    lookback_lim = len(boolean_arr) + 1

Ali_Sh 4 年前

你倒签的还不到,不是吗?以下内容应作为for循环使用:

print(np.min([np.sum(np.array(list_of_numbers) < high) + 1, lookback]))

Golden Lion 4 年前

回顾可以使用shift来完成。累计数可以用来得到一个连续的总数。查询可以用作筛选器

推荐文章

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

1 年前

unfolx · numpy数组不等式的执行时间

1 年前

mchaudh4 · 用numpy表示三对角矩阵

1 年前

Geremia · 2D NumPy数组+1D数组?

1 年前

LMC · Numpy数组布尔索引以获取包含元素

1 年前

HJA24 · 根据条件用值正向填充Numpy矩阵/掩码

1 年前

alpelito7 · 如何只获取numpy数组中每个递增值的第一次出现?

1 年前

Amarth Gûl · 找到一组向量的最近收敛点

1 年前

Mr. W · numpy.divide是函数、类还是其他什么?

1 年前

Mr. W · 为什么numpy.array在编辑内部数据时如此缓慢?

1 年前