Tạo ma trận NxN trong Python bằng list comprehension

Tạo ma trận vuông NxN với các số liên tiếp từ 1 đến N².

Ví dụ

N = 4
print("N =", N)

matrix = [list(range(1 + N*i, 1 + N*(i+1))) for i in range(N)]
print(f"Ma trận {N}x{N}:")
for row in matrix:
    print(row)

Kết quả

N = 4
Ma trận 4x4:
[1, 2, 3, 4]
[5, 6, 7, 8]
[9, 10, 11, 12]
[13, 14, 15, 16]

Giải thích

• Hàng i (0-index) chứa các số từ N*i + 1 đến N*(i+1).
• range(start, end) tạo dãy số, list() chuyển thành list.
• List comprehension ngoài tạo N hàng.
• Cách viết ngắn gọn, Pythonic để khởi tạo ma trận tuần tự.