Skip to content
Snippets Groups Projects
Commit b81bbed4 authored by Alán Muñoz's avatar Alán Muñoz
Browse files

refactor(signal): replace np -> bn

parent e2c40907
No related branches found
No related tags found
No related merge requests found
......@@ -2,6 +2,7 @@ import typing as t
from copy import copy
from pathlib import PosixPath
import bottleneck as bn
import h5py
import numpy as np
import pandas as pd
......@@ -79,7 +80,7 @@ class Signal(BridgeH5):
@staticmethod
def get_retained(df, cutoff):
return df.loc[df.notna().sum(axis=1) > df.shape[1] * cutoff]
return df.loc[bn.nansum(df.notna(), axis=1) > df.shape[1] * cutoff]
def retained(self, signal, cutoff=0.8):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment