ฟังก์ชัน map สำหรับการวนซ้ำวัตถุใน Python
สมมติว่าเรามีฟังก์ชัน square สำหรับ
ยกกำลังสองตัวเลข และมีรายการ
ที่ต้องนำฟังก์ชันนี้ไปใช้กับ
องค์ประกอบของมัน:
def square(num):
return num ** 2
lst = [2, 3, 6, 8, 15]
ใน Python เพื่อแก้ไขปัญหานี้ สามารถ
ใช้ฟังก์ชันพิเศษ map
ได้ โดยรับพารามิเตอร์เป็นฟังก์ชันและรายการ
ที่ต้องการนำฟังก์ชันไปใช้กับองค์ประกอบของมัน
ลองใช้ map เพื่อแก้ไข
ตัวอย่าง:
res = map(square, lst)
print(res)
แต่ละรายการเช่นเดียวกับวัตถุที่ซับซ้อนใด ๆ ใช้พื้นที่มากในหน่วยความจำระบบ ของ Python ดังนั้นเพื่อประหยัดทรัพยากร หลังจาก รันโค้ด จะไม่ส่งคืนรายการใหม่ แต่เป็นวัตถุ map ที่สามารถวนซ้ำได้แบบพิเศษ:
<map object at 0x000001F16674BA00>
ลองวนซ้ำมันด้วยลูป:
for el in res:
print(el)
ผลลัพธ์จะแสดงองค์ประกอบทั้งหมดของ รายการใหม่:
4
9
36
64
225
เพื่อสร้างรายการใหม่จากวัตถุ map
ต้องนำฟังก์ชัน list ไปใช้กับมัน:
lst = [2, 3, 6, 8, 15]
res = map(square, lst)
ผลลัพธ์จะแสดงรายการต่อไปนี้:
[4, 9, 36, 64, 225]
นอกจากนี้ เมื่อทำงานกับฟังก์ชัน map
ในพารามิเตอร์แรกสามารถกำหนดฟังก์ชันแลมบ์ดา
ได้ ลองเขียนตัวอย่างก่อนหน้าใหม่ด้วย
การใช้ฟังก์ชันแลมบ์ดา:
res = map(lambda num: num ** 2, lst, lst)
print(list(res))
เขียนโค้ดต่อไปนี้ใหม่ด้วยฟังก์ชันแลมบ์ดา:
def func(num):
return num + 1
lst = [1, 2, 3, 4, 5]
res = map(func, lst)
print(list(res))
เขียนโค้ดต่อไปนี้ใหม่ด้วยฟังก์ชันแลมบ์ดา:
def func(txt):
return txt[::-1]
lst = ['123', '456', '789']
res = map(func, lst)
print(list(res))