pandas获取某列前n位字符串

只需三步

• 1、指定列

book_id=df.pop('索书号')  //定义book_id获取索书号列

• 2、book_id重新取值，取索书号列第一位字符（从索引0开始取，包括第0个）

book_id=book_id.str[:1]

• 3、在第二列新增book_id列，并填充

/*用book_id填充*/
df.insert(1,'book_id',book_id)

• 完整代码

import pandas as pd
//导入数据
fpath='xx.xlsx'
df=pd.read_excel(fpath)
df.set_index("序号",inplace=True)

#指定列插入某列前n位字符串
book_id=df.pop('索书号')  //book_id获取索书号列
//  book_id重新取值，取索书号列第一位字符
book_id=book_id.str[:1]
//在第二列新增book_id列并用book_id（第一位字符）填充
df.insert(1,'book_id',book_id)
//按照book_id列对新增的”类别“列填充类别，我自定义的函数get_classify没写出（按照book_id的值，返回对应“类别”名）
df.loc[:,"类别"]=df.loc[:,'book_id'].apply(get_classify)
//输出
df.to_excel("111.xlsx")