Fix index return type of exact match and add isnan

This commit is contained in:
rasmusvt 2022-10-23 20:24:31 +02:00
parent cefd7a2edd
commit 686ef6ce28

View file

@ -174,7 +174,7 @@ def find_neighbours(value, df, colname, start=0, end=-1):
exactmatch = df[df[colname] == value] exactmatch = df[df[colname] == value]
if not exactmatch.empty: if not exactmatch.empty:
return exactmatch.index return exactmatch.index.values[0]
else: else:
lower_df = df[df[colname] < value][colname] lower_df = df[df[colname] < value][colname]
upper_df = df[df[colname] > value][colname] upper_df = df[df[colname] > value][colname]
@ -191,3 +191,8 @@ def find_neighbours(value, df, colname, start=0, end=-1):
upperneighbour_ind = np.nan upperneighbour_ind = np.nan
return [lowerneighbour_ind, upperneighbour_ind] return [lowerneighbour_ind, upperneighbour_ind]
def isnan(value):
return value!=value