读取csv写一个从B列提取“罪名”(因涉嫌XX罪)到G列的代码
时间: 2024-10-20 21:13:21 浏览: 14
python pandas读取csv后,获取列标签的方法
5星 · 资源好评率100%
在Python中,我们可以使用pandas库来处理CSV文件,包括读取数据并操作特定列。假设你的CSV文件有B列包含"罪名"格式的数据,你可以按照下面的代码示例来提取并写入新的G列:
```python
import pandas as pd
# 读取CSV文件
df = pd.read_csv('your_file.csv') # 将'yout_file.csv'替换为你的文件路径
# 假设B列名为'Sin', '罪名'在B列的前面部分,例如'B 罪名:详细描述'
# 提取"罪名"部分,这里我们简单地分割字符串,实际应用中可能会更复杂,如使用正则表达式
df['G_Sin'] = df['B'].str.extract(r'^.*?(\w+)', expand=False)
# 如果G列还未存在,创建新列
if 'G_Sin' not in df.columns:
df['G_Sin'] = ''
# 将提取后的结果写回CSV文件
df.to_csv('output_file.csv', index=False) # 将'output_file.csv'替换为你想要保存的新文件名
阅读全文