如何为已导入数据新增Plane type列，提取TAIL_NUM列前4位字符

实现步骤 #

你可以直接用pandas的字符串切片方法实现需求，操作如下：

1. 先将TAIL_NUM列转为字符串类型，避免数值类型调用字符串方法报错：

df['TAIL_NUM'] = df['TAIL_NUM'].astype(str)

2. 提取前4位字符生成Plane type列，两种写法效果一致，任选其一即可：

# 写法1
df['Plane type'] = df['TAIL_NUM'].str[:4]
# 写法2
df['Plane type'] = df['TAIL_NUM'].str.slice(stop=4)

如果需要过滤TAIL_NUM长度不足4的无效数据，可以在生成新列前增加过滤逻辑：

df = df[df['TAIL_NUM'].str.len() >=4]

操作完成后可以执行print(df[['TAIL_NUM','Plane type']].head())验证生成结果是否正确。

你上传的代码截图如下：

内容的提问来源于stack exchange，提问作者user16760437