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

หาผลรวมอาร์เรย์ 2D ใน Python ด้วยฟังก์ชัน map()

ในบทช่วยสอนนี้ เราจะหาผลรวมของอาร์เรย์ 2D โดยใช้ฟังก์ชัน map ใน Python

ฟังก์ชัน map รับสองอาร์กิวเมนต์ คือ function และ iterable มันส่งทุกองค์ประกอบของ iterable ไปยังฟังก์ชันและเก็บผลลัพธ์ไว้ใน map object เราสามารถแปลง map object ให้เป็น iterable ได้

มาดูวิธีหาผลรวมของอาร์เรย์ 2D โดยใช้ฟังก์ชัน map

  • เริ่มต้นอาร์เรย์ 2D โดยใช้รายการ

  • ส่งฟังก์ชัน sum และ 2D array ไปยังฟังก์ชัน map

  • หาผลรวมของ map object ที่ได้และพิมพ์ออกมา

ตัวอย่าง

ดูโค้ดด้านล่าง

# เริ่มต้นอาร์เรย์ 2D
array = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]
# ส่ง sum, array ไปยังฟังก์ชัน
result = list(map(sum, array))
# ดูค่าผลลัพธ์
# มันมีผลรวมของทุก sub array
print(result)

ผลลัพธ์

ถ้าคุณรันโค้ดข้างต้น คุณจะได้ผลลัพธ์ดังนี้

[6, 15, 24]

ผลลัพธ์

ตอนนี้ หาผลรวมของ result โดยใช้ฟังก์ชัน sum เดียวกัน

# หาผลรวมของ result
print(sum(result))

ผลลัพธ์

ถ้าคุณเพิ่มโค้ดสั้นข้างต้นเข้าไปในโปรแกรมข้างต้นและรัน คุณจะได้ผลลัพธ์ดังนี้

45

สรุป

หากคุณมีข้อสงสัยในบทช่วยสอนนี้ ให้ระบุในส่วนความคิดเห็น