Как я могу использовать условие if в Numpy Sum - PullRequest
0 голосов
/ 09 июля 2020

У меня есть массив, показанный ниже

arr =np.array([85,22,3,24,56,89,84,15,1,45,1,40,65])

Я хочу сделать общую сумму, которая меньше 25.

Я знаю, что могу

total_price = 0
for x in arr:
    if x < 25:
        total_price += x

, но есть ли что-то вроде sum(arr, i <25), которое вернет значение общей суммы, которое меньше 25? В С ++ есть лямбда-функция, могу ли я сделать это в python? Как я могу использовать numpy sum? лучше использовать numpy lib?

1 Ответ

2 голосов
/ 09 июля 2020

Попробуйте:

import numpy as np

arr = np.array([85,22,3,24,56,89,84,15,1,45,1,40,65])

>>> arr[arr<25].sum()

66
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...