python - 当累积和超过 0 时，列表中的先进先出配对

Question

我正在尝试实现 FIFO 配对

假设我有一个列表，其中包含 BUY/SHORT 数量和 SELL/COVER 数量，例如:

x = [100.0, -100.0, 100.0, 100.0, -200.0, 200.0, -100.0, -100.0, 100.0,-100.0]

我正在尝试隔离总和等于 0 的买入和卖出。

i.e. x[0] is offset with x[1]
x[2] and x[3] is offset with x[4]

我正在尝试获取列表 x 的索引位置的嵌套列表，如下所示:

[[[0], [1]], [[2, 3], [4]], [[5], [6, 7]], [[8], [9]]]

如果初始列表是反向的，我希望得到完全相同的结果:

x = [-100.0, 100.0, -100.0, -100.0, 200.0, -200.0, 100.0, 100.0, -100.0,100.0]

差不多，每次累计和超过 0 时，我都会重置配对。

感谢任何帮助!

我已经实现了一个部分解决方案，其中列表具有完美的偏移大小，例如:

def find_matching_position(trade_list):
    solutions = list(zip([i for i, x2 in enumerate(trade_list) if x2 == trade_list[0]],
                         [i for i, x2 in enumerate(trade_list) if x2 == trade_list[0] * -1]))

    return [sorted(x) for x in solutions]

x = [100, 100, -100 , 100, -100, -100]

    find_matching_position(x)
    [[0, 2], [1, 4], [3, 5]]

Answer 1

你可以试试这个:

x = [-100.0, 100.0, -100.0, -100.0, 200.0, -200.0, 100.0, 100.0, -100.0,100.0]
current_sum = 0
current_sub = 0
t1 = []
t2 = []
t3 = []
for i, a in enumerate(x):
   if a >= 0:
      if (current_sum + a)+current_sub == 0:
          t1.append(i)
          t2.append([t1, t3])
          t1 = []
          t3 = []
          current_sum = 0
          current_sub = 0
       else:
          current_sum += a
          t1.append(i)
   else:
       if current_sum + (current_sub + a) == 0:
           t3.append(i)
           t2.append([t1, t3])
           t1 = []
           t3 = []
           current_sum = 0
           current_sub = 0
       else:
           current_sub += a
           t3.append(i)

print(t2)

输出:

[[[0], [1]], [[2, 3], [4]], [[5], [6, 7]], [[8], [9]]]