Среднее Numpy с условием

У меня есть алгоритм расчета средней скорости на чистом питоне:

    speed = [...]
    avg_speed = 0.0
    speed_count = 0
    for i in speed:
        if i > 0: # I dont need zeros
            avg_speed += i
            speed_count += 1

    if speed_count == 0:
        return 0.0

    return avg_speed / speed_count

Есть ли способ переписать эти функции с помощью Numpy?

9
задан Artem Mezhenin 18 June 2012 в 14:13
поделиться