Make results retrievable per row index

This commit is contained in:
reedts 2018-05-29 15:17:49 +02:00
parent 7f843ed051
commit cebed01f35

View file

@ -1,3 +1,4 @@
import pandas as pd
from peewee import * from peewee import *
from playhouse import sqlite_ext from playhouse import sqlite_ext
from playhouse.sqlite_ext import SqliteExtDatabase from playhouse.sqlite_ext import SqliteExtDatabase
@ -27,3 +28,23 @@ class Result(Model):
class Meta: class Meta:
database = db database = db
def rows_to_df(indices):
temps = []
dcs = []
winds = []
db.connect()
for result in Result.select().where(Result.id << indices):
temps += result.temperature
dcs += result.dc_output
winds += result.wind_speed
db.close()
return pd.DataFrame(
{'temp': temps,
'dc': dcs,
'wind': winds
})