我试图用python编写一个语句来转换这个输入(例如):
[[3,4],[1],[1,2]]
在这个输出中:
[3,4,-,1,-,1,2]
仅使用zip和列表理解
这是我的代码:
a = [[1,2],[1],[3,4]]
result = [j for i in zip(a,'-'*len(a)) for j in i]
print(result)
但我得到的只是:
[[1, 2], '-', [1], '-', [3, 4], '-']
我的输出应该是:
[1, 2, '-', 1, '-', 3, 4]
我错过了什么？
这应该是有效的:
a = [[1,2],[1],[3,4]]
result = [a for i in zip(a,'-'*len(a)) for j in i for a in j]
print(result)
输出:
[1, 2, '-', 1, '-', 3, 4, '-']
如果你不想把最后'-'个也包括在内，这也很有效:
a = [[1,2],[1],[3,4]]
result = [a for i in zip(a, '-'*len(a)) for j in i for a in j][:-1]
print(result)
输出:
[1, 2, '-', 1, '-', 3, 4]