作为对一个平时靠c++吃饭的程序员来说,偶尔也想写点python程序,要想写就得要学,要学就得要总结和关联,把要学的知识与已学的知识相关联起来,可以更快地学习和理解新的知识。
前面看过了reduce函数,今天来看一下map函数,map函数是python的内置函数,介绍如下:
map() 函数会根据提供的函数对指定序列做映射。
第一个参数 function 以参数序列中的每一个元素调用 function 函数,返回包含每次 function 函数返回值的新列表。
语法
以下是 map() 方法的语法:
map(function, iterable, ...)
参数
- function -- 函数
- iterable -- 一个或多个序列
先上一个例子:
print(list(map(lambda x : x*x, [1,2,3,4,5])))
print(list(map(lambda x, y: x*y, [1,2,3,4,5], [6,7,8,9,10])))
输出:[1, 4, 9, 16, 25]
[6, 14, 24, 36, 50]
对比第一个例子,其实std::for_each貌似勉强可以凑合着去满足一下,std::for_each有返回值,但是返回的是一个函数对象,不太好搞啊。
更符合一点的函数时std::transform,这个函数中带有输出序列的迭代器,可以满足输入是一个容器和两个容器的情况,超过两个输入std::transform就不行了,但是python的map还行,这个很厉害了,比如:
print(list(map(lambda x, y, z: x*y+z, [1, 2, 3, 4, 5], [1, 2, 3, 4, 5], [1, 2, 3, 4, 5])))
看看std::transform的吧
std::vector<int> input1{1, 2, 3, 4, 5};
std::vector<int> input2{6, 7, 8, 9, 10};
std::vector<int> output1;
std::vector<int> output2;
std::transform(input1.begin(), input1.end(), std::back_inserter(output1),
[](int x) { return x*x; });
std::transform(input1.begin(), input1.end(), input2.begin(), std::back_inserter(output2),
[](int x, int y) { return x*y; });
当然c++也可以直接修改容器本身而无需创建新的容器,这个也很实用
std::vector<int> input1{1, 2, 3, 4, 5};
std::for_each(input1.begin(), input1.end(), [](int& n) { n*=3; });
std::transform(input1.begin(), input1.end(), input1.begin(), [](int x) { return x/3; });
看上去这个map函数比reduce函数容易理解一点,单个函数的能力总是有限的,武器组合起来就能产生无穷的威力。
在HackerRank上有这么一道题,要求输入一个整数,打印出如下图形(输入为5的时候):
1
121
12321
1234321
123454321
此题目要求不能超过2行代码,网络上有人用1行代码做出了解答,包含了 reduce函数和map函数,解法如下:
for i in range(1,int(input())+1): print(reduce(lambda x,y:x+y, list(map(str, (list(range(1,i))+list(range(i,0,-1)))))))
看上去是不是很不错!
用c++也来试试,一行两行的解法不会,试试简单的写法,不就是行数多一点嘛:
for (int i = 1; i <= 5; ++i) {
std::vector<int> vec(2 * i - 1);
std::iota(vec.begin(), vec.begin() + i, 1);
std::reverse_copy(vec.begin(), vec.begin() + i - 1, vec.begin() + i);
std::copy(vec.begin(), vec.end(), std::ostream_iterator<int>(std::cout));
std::cout << std::endl;
}
反正我喜欢!