python实现二维列表的列表生成式

[[expression for variable in iterable] for variable in iterable]

identity_matrix = [[1 if i == j else 0 for i in range(3)] for j in range(3)]
print(identity_matrix)
# 输出:
# [[1, 0, 0],
#  [0, 1, 0],
#  [0, 0, 1]]

multiplication_table = [[i * j for i in range(1, 5)] for j in range(1, 5)]
print(multiplication_table)
# 输出:
# [[1, 2, 3, 4],
#  [2, 4, 6, 8],
#  [3, 6, 9, 12],
#  [4, 8, 12, 16]]

diagonal_matrix = [[1 if i == j or i + j == 4 else 0 for i in range(5)] for j in range(5)]
print(diagonal_matrix)
# 输出:
# [[1, 0, 0, 0, 1],
#  [0, 1, 0, 1, 0],
#  [0, 0, 1, 0, 0],
#  [0, 1, 0, 1, 0],
#  [1, 0, 0, 0, 1]]

# 创建二维列表
lst = [
    ['城市', '环比', '同比'],
    ['北京', 102, 103],
    ['上海', 104, 504],
    ['深圳', 100, 39]
]
print(lst)

for row in lst:
    for item in row:
        print(item, end='\t')
    print()

# 列表生成式生成一个4行5列的二维列表
lst2 = [[j for j in range(5)] for i in range(4)]
print(lst2)