如何根据列表的值拆分 python 中的列表

Question

我有一个包含数字子列表的列表，并且想要提取特定的数字。 在我的简化示例中，我有两个主要子列表，每个子列表都有自己的一对数字：

data=[[[1, 0], [2, 0], [2, 1], [2, 2],\
       [1, 0], [1, 1], [1, 2],\
       [0, 1], [0, 2], [0, 3]],\
      [[1, 0], [2, 0],\
       [1, 0],\
       [0, 1], [0, 2], [1, 2],\
       [1, 0], [1, 1], [1, 1]]]

存储在data中的对可以根据一些规则进行划分，我想要每个划分的最后一对。 为简单起见，我将每个部门显示为data中的一行。 每个除法以[1, 0]或[0, 1]开头，这两对是断点。 然后，我只是想要每个断点之前的最后一对。 如果我可能在两个断点之间没有点，我只导出前一个断点。 最后我想要它作为以下列表：

data=[[[2, 2],\
       [1, 2],\
       [0, 3]],\
      [[2, 0],\
       [1, 0],\
       [1, 2],\
       [1, 1]]]

Answer 1

data=[[[1, 0], [2, 0], [2, 1], [2, 2],
   [1, 0], [1, 1], [1, 2],
   [0, 1], [0, 2], [0, 3]],
  [[1, 0], [2, 0],
   [1, 0],
   [0, 1], [0, 2], [1, 2],
   [1, 0], [1, 1], [1, 1]]]
newData = []

for subarray in data:
    new_subarray = []
    for i,item in enumerate(subarray):
        if item == [0,1] or item == [1,0]:
            if i> 0:
                new_subarray.append(subarray[i-1])
        if i == len(subarray)-1:
            new_subarray.append(item)
    newData.append(new_subarray)

print(newData)

Answer 2

您可以使用enumerate执行以下操作：

def fun(lst):
    return [p for i, p in enumerate(lst) if i==len(lst)-1 or set(lst[i+1])=={0,1}]
 
[*map(fun, data)]
# [[[2, 2], [1, 2], [0, 3]], [[2, 0], [1, 0], [1, 2], [1, 1]]]

fun过滤所有元素的嵌套列表，这些元素要么是最后一个，要么是[0, 1]或[1, 0] 。

Answer 3

这是一个有趣的小不可读 numpy oneliner：

import numpy as np
[np.array(a)[np.roll(np.flatnonzero(np.logical_or(np.all(np.array(a)==(1, 0), axis=1), np.all(np.array(a)==(0, 1), axis=1)))-1, -1)].tolist() for a in data]

# [[[2, 2], [1, 2], [0, 3]], [[2, 0], [1, 0], [1, 2], [1, 1]]]

它有效，但实际上您最好使用schwobaseggl的解决方案。