Selaa lähdekoodia

Further optimisations

Petra Lamborn 5 vuotta sitten
vanhempi
commit
88989c825a
2 muutettua tiedostoa jossa 5 lisäystä ja 2 poistoa
  1. 4
    1
      py/downkwh.py
  2. 1
    1
      py/util.py

+ 4
- 1
py/downkwh.py Näytä tiedosto

19
                      '2017-{:02d}-01 00:30:00'.format(mstarts[i]), 
19
                      '2017-{:02d}-01 00:30:00'.format(mstarts[i]), 
20
                      '2017-{:02d}-01 00:00:00'.format(mends[i]), 
20
                      '2017-{:02d}-01 00:00:00'.format(mends[i]), 
21
                      '%%1')
21
                      '%%1')
22
+    print("Pivoting")
23
+    kwhpiv = kwhdata.pivot(index = 'read_time', columns = 'icp_id', values = 'kwh_tot')
22
     print("Pickling")
24
     print("Pickling")
23
-    kwhdata.to_pickle('../data/2017-{}-all.pkl'.format(m))
25
+    kwhpiv.to_pickle('../data/2017-{}-all.pkl'.format(m))
24
     del kwhdata
26
     del kwhdata
27
+    del kwhpiv
25
     gc.collect()
28
     gc.collect()
26
 
29
 
27
 print('Done')
30
 print('Done')

+ 1
- 1
py/util.py Näytä tiedosto

116
     qdf = getQuery(query, pdict)
116
     qdf = getQuery(query, pdict)
117
     print("Optimising")
117
     print("Optimising")
118
     qdf['icp_id'] = qdf['icp_id'].astype(np.int32)
118
     qdf['icp_id'] = qdf['icp_id'].astype(np.int32)
119
-    qdf['kwh_tot'] = qdf['kwh_tot'].astype(np.float32)
119
+    qdf['kwh_tot'] = qdf['kwh_tot'].astype(np.float16)
120
     print("Done")
120
     print("Done")
121
     return(qdf)
121
     return(qdf)
122
 
122