fz0x1/HPI
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

my.body.sleep: massive speedup for average temperature calculation

Browse source
This commit is contained in:
Dima Gerasimov 2023-11-11 00:28:13 +00:00 • committed by karlicoss
parent 37643c098f
commit bde43d6a7a
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
my/body/sleep/common.py
View file

@ -17,15 +17,21 @@ class Combine:
bdf = BM.dataframe() bdf = BM.dataframe()
temp = bdf['temp'] temp = bdf['temp']
# sort index and drop nans, otherwise indexing with [start: end] gonna complain
temp = pd.Series(
temp.values,
index=pd.to_datetime(temp.index, utc=True)
).sort_index()
temp = temp.loc[temp.index.dropna()]
def calc_avg_temperature(row): def calc_avg_temperature(row):
start = row['sleep_start'] start = row['sleep_start']
end = row['sleep_end'] end = row['sleep_end']
if pd.isna(start) or pd.isna(end): if pd.isna(start) or pd.isna(end):
return None return None
between = (start <= temp.index) & (temp.index <= end)
# on no temp data, returns nan, ok # on no temp data, returns nan, ok
return temp[between].mean() return temp[start: end].mean()
df['avg_temp'] = df.apply(calc_avg_temperature, axis=1) df['avg_temp'] = df.apply(calc_avg_temperature, axis=1)
return df return df