使用str.extract()函数和正则表达式来切分

xujunzju • 2023年7月7日 00:40 • 编程语言 • 阅读 160

# 创建数据集
data = {
    '姓名': ['沈文辉', '张金顺', '吴桂英', '吴桂英', '吴桂英', '吴桂英', '吴桂英', '王伯忠', '王伯忠', '王伯忠'],
    '第二列': ['阴性', '阴性', '嗜麦芽窄食单胞菌2803', '肺炎克雷伯菌79', '粪肠球菌25', '白色念珠菌53', '人类疱疹病毒一型11082', '铜绿假单胞菌258', '肺炎克雷伯菌771', '嗜麦芽窄食单胞菌653']
}

# 创建DataFrame
df = pd.DataFrame(data)

你可以使用str.extract()函数和正则表达式来实现这一目标。先创建一个新的列”A”，使用正则表达式提取出第二列中的文字部分，如果没有文字则填空。再创建一个新的列”B”，使用正则表达式提取出第二列中的数字部分，如果没有数字则填空。


# 提取文字部分
df['A'] = df['第二列'].str.extract('([^\d]+)', expand=False).str.strip()

# 提取数字部分
df['B'] = df['第二列'].str.extract('(\d+)', expand=False)

# 填充空值
df.fillna('', inplace=True)

print(df)

运行以上代码，输出结果如下：