Computer >> บทช่วยสอนคอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

สร้าง Matrix ขนาด N×N ใน Python ด้วย List Comprehension


สร้าง Matrix สี่เหลี่ยมจัตุรัส N×N ที่มีตัวเลขเรียงต่อเนื่อง 1 ถึง N² โดยใช้ List Comprehension

ตัวอย่างโค้ด

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

matrix = [list(range(1 + N * i, 1 + N * (i + 1))) for i in range(N)]
print("N×N Matrix:")
for row in matrix:
    print(row)

ผลลัพธ์

N = 4
N×N Matrix:
[1, 2, 3, 4]
[5, 6, 7, 8]
[9, 10, 11, 12]
[13, 14, 15, 16]

คำอธิบาย

  • แถว i (0-based) เริ่มที่ 1 + N*i ถึง N*(i+1)

  • range(start, end) สร้างลำดับต่อเนื่อง

  • list(...) แปลงเป็นลิสต์แถว

  • List Comprehension วน i = 0 ถึง N-1 สร้างทุกแถว