File tree 1 file changed +5
-5
lines changed
py-hftbacktest/hftbacktest/data/utils
1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -120,10 +120,10 @@ def convert_snapshot(
120
120
121
121
ss_bid = ss_bid [:ss_bid_rn ]
122
122
ss_ask = ss_ask [:ss_ask_rn ]
123
- snapshot = np .zeros (len (ss_bid ) + len (ss_ask ), event_dtype )
123
+ snapshot = np .empty (len (ss_bid ) + len (ss_ask ), event_dtype )
124
124
125
- snapshot += [ cols for cols in sorted (ss_bid , key = lambda v : - float (v [4 ]))]
126
- snapshot += [ cols for cols in sorted (ss_ask , key = lambda v : float (v [4 ]))]
125
+ snapshot [: len ( ss_bid )] = sorted (ss_bid , key = lambda v : - float (v [4 ]))
126
+ snapshot [ len ( ss_bid ): len ( ss_bid ) + len ( ss_ask )] = sorted (ss_ask , key = lambda v : float (v [4 ]))
127
127
128
128
if output_filename is not None :
129
129
np .savez (output_filename , data = snapshot )
@@ -271,7 +271,7 @@ def convert(
271
271
qty = float (row [qty_col ])
272
272
273
273
# Insert TRADE_EVENT
274
- tmp [row_num ] = [
274
+ tmp [row_num ] = (
275
275
TRADE_EVENT | (SELL_EVENT if row [side_col ] else BUY_EVENT ), # trade initiator's side
276
276
exch_ts ,
277
277
local_ts ,
@@ -280,7 +280,7 @@ def convert(
280
280
0 ,
281
281
0 ,
282
282
0
283
- ]
283
+ )
284
284
row_num += 1
285
285
tmp = tmp [:row_num ]
286
286
You can’t perform that action at this time.
0 commit comments