0%

Python如何处理单行输入的多个数字,以及map函数的使用

这段时间参加了几个公司的在线笔试,发现了一个很有意思的现象,因为不同公司使用的测试平台不同,所以将输入的数据告诉代码的方式也会有所区别。一种是“保姆”类型,即作为应试者,并不需要考虑如何将输入转换成机器能看得懂的代码,而只要实现解决问题的特定函数就可以了。输入的话已经有固定的代码实现了。典型例子比如牛客、Leetcode以及HackerRank。另外一种是应试者需要考虑如何将输入转变为代码的全过程,这就有些像本科时做的那种Online Judge,给个题目,然后编程板里全是空白的,需要从零开始写起。

比如说有这样的输入样例,其中第一行代表总共有多少数字,而第二行的输入代表每个数字的内容。

5
1 2 3 4 5

这种的话用C++之类的语言就非常好实现。只需要定义好对应变量的类型,跟着输入样例的顺序进行数据输入的实现就可以了。

int n;
cin >> n;
vector<int> array;
for(i = 0; i < n; i++){
    int element;
    cin >> element;
    array.push_back(element);
}

但是如果使用的是Python环境的话就会出一些问题,比如我们把上面的代码修改为python的形式:

n = int(input());
array = []
for i in range n:
    int element;
    element = int(input());
    array.append(element);

这个时候如果我们按照样例输入的方式输入数据,Python会在element的部分报错,原因是无法将输入转换成int格式的内容。
为什么会出现这样的原因呢?其实是因为当我们输入一行数据的时候,python会直接将这一整行的数据赋值给element,所以当第一次输入element的时候,最后得到的element是”1 2 3 4 5”的字符串格式(没进行强制int转换的情况下),因为中间有空格所以无法转换成数字(就算省去空格也不是我们想要的数字)。
像C++这样的语言上述代码可以跑的一个很重要的原因是它们在定义每个变量的时候,会特定指出这个变量是哪个类型的,因此输入的时候空格就相当于一个截断操作,来进行数字和数字的区分。(因为int类型不可能会包含空格)
那么在Python中,应该如何设置才可以在只用一行输入的情况下,完成一个数组的赋值呢?
首先我们需要放弃使用For循环,因为Python是将整行输入赋值给对应变量的,那么如果用for循环,即使输入没有报错,我们也没有n行的输入来完成这个循环.(因为我们只有一行)
其次我们需要使用split函数来切分输入的这一行数据,使其变成[“1”,”2”,”3”,”4”,”5”]的形式。

element = input().split()

这个时候需要注意的是,虽然我们已经得到了一个数组,但是里面的每个元素都是str字符串格式,而不是我们想要的int格式,那么这个时候需要如何操作呢?
此时可以使用到python的map函数,它的主要作用是对迭代对象进行对应的映射,并返回一个迭代器,它的用法如下,

map(function, iterable, …)

第一个参数是function,即需要进行映射的函数。第二个参数是需要进行映射的可迭代对象(iterable),例如list就是一种可迭代对象。
举个栗子,假设我们定义一个函数,使其返回值是输入值的2倍,那么我们可以这么写,

def a(num):
    return num * 2

那么这个函数a就可以作为第一个参数function的输入,与此同时我们定义一个输入的列表array,

array = [1, 2, 3, 4, 5]  

这样我们可以通过map生成一个对应的迭代器。当然迭代器自己是无法做输出的(如果直接print也是只会显示迭代器自己的信息),这个时候可以使用list()函数将迭代器转化为列表。

array = [1, 2, 3, 4, 5]  
print(list(map(a, array)))

这时我们可以看到输出为[2, 4, 6, 8, 10],这样就完成了一个映射的全过程。其实map函数是一个不用for循环来实现批量操作的很好的方法。
回到之前的问题,我们应该如何将字符串数组变成一个纯数字类型的数组呢?这个时候我们可以通过map映射对字符串数组中的每个元素做一个int强制转换,即调用int()函数。然后通过list()将生成的迭代器转换为数组。

print(list(map(int, input().split())))

如果我们的输入为1 2 3 4 5, 那么对应的输出就变为[1, 2, 3, 4, 5]了。这样我们就可以实现将一行的多个数据转变为一个数组的功能了。
完整代码如下:

element = list(map(int, input().split()))
# finish your input
print(element)