Как получить индекс отсортированного массива, в котором сумма первых n значений больше определенного значения

Question

если я получил отсортированный массив N = 0, 1, 2, 3, 4, 5, 6], я хочу получить индекс, в котором сумма A [0: index] больше определенного числа (предполагается чтобы быть 9, в этом случае значение индекса должно быть 3), как я могу написать его на Python?

Я думаю, что я могу использовать np.where () или np.cumsum (), но это не работает

np.argwhere(np.cumsum(A) > 9)

конечно, это неправильно, так какой же правильный метод?

Answer 1

Я предлагаю:

def foo(array, threshold):
    valid_entries = np.cumsum(array) > threshold
    return np.argmax(valid_entries) if np.any(valid_entries) else None

Answer 2

Вы были близки. Вы можете просто использовать argmax следующим образом. Кроме того, при использовании argmax добавьте проверку, чтобы убедиться, что значение по крайней мере 1 пересекает порог.

threshold = 9
temp = np.cumsum(A) > threshold

if temp.any():
    print(np.argmax(temp))
else:
    print("Threshold never crossed")