主页 / 问题 / 1436179
Accepted
Nommad
Asked:2022-08-04 21:40:28 +0800 CST2022-08-04 21:40:28 +0800 CST

在 zip() 之后转换列表

  • 772
# изначально был список
list1 = [[0,0,1],[0,0,1],[0,0,1],[2,2],[1,2,3,4]]
result = []
    for element in list1:
        if len(element )!=2:
            result.append([[x, y] for x,y in zip(element , element[1:])])
        else:
            result.append(element )
print(result)
#

结论

[[[0, 0], [0, 1]], 
[[0, 0], [0, 1]], 
[[0, 0], [0, 1]], 
[2, 2], 
[[1, 2], [2, 3], [3, 4]]]

期望的输出

[[0,0,1],[0,0,1],[0,0,1],[2,2],[1,2,3,4]]

请告诉我，我在输入时得到这样的列表。我怎样才能把它恢复到原来的格式？在我申请之前zip()。以防万一，我会澄清list1每次都不一样。它是通过 psycopg 从请求中获取的。也就是说，简单地删除一些索引是行不通的。理想情况下是某种验证。我尝试使用len()，但事实证明列表列表中的元素数量[[0, 0], [0, 1]], [2, 2]相同且等于2

python

1 个回答

  1. Best Answer

    isinstance定义列表的第一个元素的类型：

    def connect_one(lst):
    first = lst[0]
    if isinstance(first, list):
        return [first[0]] + [sublist[-1] for sublist in lst]
    return lst


def connect(lst):
    return list(map(connect_one, lst))


print(connect([
    [[0, 0], [0, 1]], 
    [[0, 0], [0, 1]], 
    [[0, 0], [0, 1]], 
    [2, 2], 
    [[1, 2], [2, 3], [3, 4]]
]))

    $ python connect.py
[[0, 0, 1], [0, 0, 1], [0, 0, 1], [2, 2], [1, 2, 3, 4]]
    • 4

相关问题