refactored and moved count_true from df to seq utils

This commit is contained in:
2020-11-04 15:32:11 +01:00
parent 1d96ec18d8
commit ea2293429e
3 changed files with 9 additions and 8 deletions

View File

@ -38,10 +38,4 @@ def drop_col(df, name):
df.drop(name, axis="columns", inplace=True)
def count_true(bdf):
good = bdf[bdf]
ngood = len(good)
return ngood

View File

@ -1,6 +1,5 @@
from .consts import SYM_NOTHING, SYM_GOOD, SYM_BAD
from .df import count_true
from .seq import is_empty
from .seq import count_true, is_empty
def print_outcome(state, units):

View File

@ -1,5 +1,13 @@
import numpy as np
def count_true(seq):
bools = np.asanyarray(seq, dtype=bool)
return np.count_nonzero(bools)
def is_empty(seq):
return len(seq) == 0