需求
今儿有朋友问 老Amy ,要怎么用 python 快速的实现以下 需求:
在这里,我先确认是否只提取出前两位。毕竟如果有些县城的名字是三个字,那写法可能就需要更复杂一丢丢了。但是,这位朋友坚决的说只有两个字…emmm…细想了一下,好像目前为止所听到县城的名字都是两个字奥~巧…hhh
那这就 sosososo easy 了
思路
- 读取该列
- 获取前两个字符
- 添加至新列。就完成了~
实现
构造伪数据
import numpy as np
import pandas as pd
data = {
"info":["望城区一楼机房","岳麓区二楼机房","开福区三楼机房"]
}
df = pd.DataFrame(data)
df
获取info
列前两个字,添加至新列中
在这里,其实我们需要对info
中每个元素进行操作,所以我们使用到pandas
中Series
对象的apply
方法,将info
列中的每个元素映射到f
方法中进行操作,然后返回。
f = lambda x:x[:2]
df["district"] = df["info"].apply(f)
df
就完成了鸭~请一健三联,hhhhh