python提取csv列中的数字作为新的一列
时间: 2024-05-06 10:19:23 浏览: 45
你好!关于你的问题,我可以给你以下的Python代码来提取csv列中的数字作为新的一列:
```python
import csv
with open('input.csv', 'r') as csv_file:
csv_reader = csv.reader(csv_file)
# 获取列标
header = next(csv_reader)
header.append('new_column')
with open('output.csv', 'w', newline='') as new_file:
csv_writer = csv.writer(new_file)
# 写入新的列标
csv_writer.writerow(header)
for row in csv_reader:
# 获取数据
old_data = row[0]
# 提取数字
new_data = ''.join(filter(str.isdigit, old_data))
# 写入新的列
row.append(new_data)
# 写入整行数据到 output.csv 文件
csv_writer.writerow(row)
```
你可以将 `input.csv` 替换为你的 csv 文件名,代码会将提取后的数据写入到 `output.csv` 文件。希望对你有所帮助!
阅读全文