python里的map函数替代循环的范例演示

2022-03-11 00:00:00 函数演示范例

map函数有的时候可以用来替代循环，列入要对列表内的数据进行批量算术运算，对两个长度相等的列表进行累加等等，使用map可以大大简化代码的写法，注意python3的map函数返回的是map对象，并非列表，所以输出的时候需要进行下转换。

"""
作者：皮蛋编程(https://www.pidancode.com)
创建日期：2022/3/19
功能描述：python里的map函数替代循环的范例演示
"""


def add1(a):
    return a + 1


def add2(a, b):
    return a + b


def add3(a, b, c):
    return a + b + c


a1 = [1, 2, 3, 4, 5]
a2 = [1, 2, 3, 4, 5]
a3 = [1, 2, 3, 4, 5]

# 对a1列表的数字全部加1
b = map(add1, a1)
print(list(b))

# 将a1和a2的数字对应相加
b = map(add2, a1, a2)
print(list(b))

# 对a1，a2，a3的数字对应相加
b = map(add3, a1, a2, a3)
print(list(b))

输出结果如下：
[2, 3, 4, 5, 6]
[2, 4, 6, 8, 10]
[3, 6, 9, 12, 15]

以上代码在Python3.9环境下测试通过

相关文章