函数介绍
描述
map() 会根据提供的函数对指定序列做映射。 第一个参数 function 以参数序列中的每一个元素调用 function 函数,返回包含每次 function 函数返回值的新列表。
语法
map() 函数语法: map(function, iterable, ...)
参数
function -- 函数 iterable -- 一个或多个序列
返回值
Python 2.x 返回列表。 Python 3.x 返回迭代器。
实例
以下实例展示了 map() 的使用方法:
>>>def square(x) : # 计算平方数
... return x ** 2
>>> map(square, [1,2,3,4,5]) # 计算列表各个元素的平方
[1, 4, 9, 16, 25]
>>> map(lambda x: x ** 2, [1, 2, 3, 4, 5]) # 使用 lambda 匿名函数
[1, 4, 9, 16, 25]
# 提供了两个列表,对相同位置的列表数据进行相加
>>> map(lambda x, y: x + y, [1, 3, 5, 7, 9], [2, 4, 6, 8, 10])
[3, 7, 11, 15, 19]
应用数据清洗
假设我们拿到下面的一份CSV数据:
序号 | 城市 |
---|---|
1 | 中国上海 |
2 | 中国北京 |
3 | 澳大利亚 |
4 | 美国 |
我们目的是将中国的国家和城市提取出来分成两列,然后外国的国家和城市不变。怎么实现呢?这就用到我们刚刚将的map()函数啦,将城市列作用于我们定义的函数,一起来看看怎么实现?
# 城市: 提取国家和城市
def transform_country(x):
if '中国' in x:
return '中国'
else:
return x
def transform_city(x):
if '中国' in x:
return x[2:]
else:
return x
df['国家'] = df.城市.map(lambda x: transform_country(x))
df['城市'] = df.城市.map(lambda x: transform_city(x))
最后的目标结果变成这样:
序号 | 国家 | 城市 |
---|---|---|
1 | 中国 | 上海 |
2 | 中国 | 北京 |
3 | 澳大利亚 | 澳大利亚 |
4 | 美国 | 美国 |
今天这个小小数据清洗方法掌握了没?每天进步一点点就是成功。
更多推荐
干货|Python学习笔记(三)字典
干货 | 50题带你入门Python数据分析(上)
干货 | 50题带你入门Python数据分析(下)长按二维码
关注我们吧