sander/rowsandall
Private
Public Access
1
0
Fork 0
Code Activity

some more polars, passing test_uploads

Browse Source
This commit is contained in:
Sander Roosendaal
2024-04-18 08:31:08 +02:00
parent e510591dba
commit eb44623e24
7 changed files with 68 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
rowers/utils.py
View File

@@ -6,6 +6,7 @@ from django.utils import timezone
import math
import numpy as np
import pandas as pd
import polars as pl
import colorsys
from django.conf import settings
import collections
@@ -342,22 +343,22 @@ def isbreakthrough(delta, cpvalues, p0, p1, p2, p3, ratio):
pwr *= ratio
delta = delta.astype(int, errors='ignore').values
cpvalues = cpvalues.astype(int, errors='ignore').values
pwr = pwr.astype(int, errors='ignore').values
delta = delta.cast(pl.Int32)
cpvalues = cpvalues.cast(pl.Int32)
pwr = pwr.cast(pl.Int32)
res = np.sum(cpvalues > pwr+1)
res2 = np.sum(cpvalues > pwr2+1)
btdf = pl.DataFrame({
'delta': delta,
'cpvalues': cpvalues,
'pwr': pwr,
'pwr2': pwr2
})
btdf = pd.DataFrame(
{
'delta': delta[cpvalues > pwr],
'cpvalues': cpvalues[cpvalues > pwr],
'pwr': pwr[cpvalues > pwr],
}
)
res = btdf.select(pl.col("cpvalues")>pl.col("pwr")+1)['cpvalues'].sum()
res2 = btdf.select(pl.col("cpvalues")> pl.col("pwr2")+1)['cpvalues'].sum()
btdf.sort_values('delta', axis=0, inplace=True)
btdf = btdf.filter(pl.col("cpvalues")>pl.col("pwr"))
btdf = btdf.sort('delta')
return res >= 1, btdf, res2 >= 1