python 同时打印两个列表中的元素,列表长度不足的一方使用 ‘-’,补齐

# 自己实现
a = [1, 2, 3, 4]
b = [1, 2, 3, 4, 6, 7]

f = lambda arr, n: arr[n] if len(arr) > n else '-'
loop_num = max(len(a), len(b))

for n in range(loop_num):
    print(f(a, n), f(b, n))
 
# 输出
1 1
2 2
3 3
4 4
- 6
- 7
# 使用 zip 函数实现
from itertools import chain, repeat
a = (1, 2, 3, 4)
b = (4, 5, 6)
# 当b序列比a序列短时,自动填充 '-' 补齐
c = zip(a, chain(b, repeat('-')))
print(list(c))
# [(1, 4), (2, 5), (3, 6), (4, '-')]
# https://www.python.org/dev/peps/pep-0201/

这种会造成python无限补齐

zip(chain(a, repeat('-')), chain(b, repeat('-'))):

同时打印两个列表中的元素,列表长度不足的一方使用 ‘-’,补齐

你可能感兴趣的:(python)