更新时间:2022-05-02 08:55:36
You can tell Python's split()
to stop after a given number of matches by passing a maxsplit
parameter. So in your case you just need to split after the first space as follows:
import csv
with open('input.csv', 'rb') as f_input, open('output.csv', 'wb') as f_output:
csv_output = csv.writer(f_output, delimiter=';')
for row in csv.reader(f_input, delimiter=';'):
# Skip empty lines
if len(row) > 3:
try:
name, surname = row[2].split(' ', 1)
except ValueError as e:
# No surname
name, surname = row[2], ''
row[2] = name
row.insert(3, surname)
csv_output.writerow(row)
因此对于具有以下内容的输入:
So for an input with:
data;data;name surname1 surname2;data;data
data;data;name surname;data;data
data;data;name surname;data;data
data;data;name surname;data;data
您会得到:
data;data;name;surname1 surname2;data;data
data;data;name;surname;data;data
data;data;name;surname;data;data
data;data;name;surname;data;data