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)
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,711评论 5 468
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,932评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,770评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,799评论 1 271
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,697评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,069评论 1 276
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,535评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,200评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,353评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,290评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,331评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,020评论 3 315
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,610评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,694评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,927评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,330评论 2 346
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,904评论 2 341

推荐阅读更多精彩内容