如果二维列表中有空元素，则将 Python 二维列表转换为 3D 列表

Question

在有或没有 NumPy 的情况下使用 Python3，我该如何转换

a =[[1,2],[0,1],[],[2,2],[],[2,13],[1,4],[6,1],[],[2,7]]

（关于它的空元素），到

b=[[[1, 2], [0, 1]], [[2, 2]], [[2, 13], [1, 4], [6, 1]], [[2, 7]]] 

我很感激这方面的任何帮助。

谢谢

Answer 1

请尝试以下脚本：

a =[[],[1,2],[0,1],[],[2,2],[],[2,13],[1,4],[6,1],[],[2,7]]
newlist=[]
temp=[]
for element in a:
    if element: # if element is not empty add to temp
        temp.append(element)
    elif temp: # if element is empty and temp is not empty add temp to newlist
        newlist.append(temp)
        temp=[]
if temp: # if temp is not empty add to newlist
    newlist.append(temp)        
print(newlist) 

Answer 2

使用itertools's groupby

from itertools import groupby

lst = []

for i, g in groupby(a, key=lambda x: len(x) >= 1):
    grp = list(g)
    if i and len(grp) >= 1:
        lst.append(grp)

---

第一：

[[[1, 2], [0, 1]], [[2, 2]], [[2, 13], [1, 4], [6, 1]], [[2, 7]]]

Answer 3

你可以这样做

def func(a):
    b = [[]]
    for i in (a): # iterate over a
        if i: # if ith element is not empty list
            b[-1].append(i) # add it to the latest list in b
        else: # otherwise append new list to b
            b.append([])
    return b

或类似地，

def func(a):
    b, c = [], []
    c = [b[-1].append(i) for i in ([[]] + a ) if (i or b.append([]))]
    return b

两者都遵循基本相同的过程，

第二个片段浪费了一个额外的变量

Answer 4

你可以试试这个。

from collections import deque

a = [[1,2],[0,1],[],[2,2],[],[2,13],[1,4],[6,1],[],[2,7]]

d = deque(a)
if d[0] == []:
    d.popleft()
if len(d) > 0:
    if d[-1] == []:
        d.pop()
tmp = []; b = []
for x in d:
    if x == []:
        b.append(tmp)
        tmp = []
    else:
        tmp.append(x)
b.append(tmp)     
print(b)


 

输出：

[[[1, 2], [0, 1]], [[2, 2]], [[2, 13], [1, 4], [6, 1]], [[2, 7]]]