python - 根据 csv 中的空格分隔值写入新行

Question

我有一个如下所示的 csv 文件:

column1 column2   column3                        
  a       b       value1 value2 value3
  a       z       value4 value5 value6
  s       t       value7  

我想用空格分隔“column3”，以便该列只有一个值。该行的其余部分将被重复。 这是我想要得到的输出示例:

column1 column2    column3  
  a       b        value1 
  a       b        value2 
  a       b        value3

请注意，我找到了类似问题的答案，但使用的是我不想使用的 pandas。

任何解决方案都会有帮助。

谢谢

Answer 1

使用csv模块。

例如:

import csv

data = []
with open(filename) as csvfile:              #Read SRC
    reader = csv.reader(csvfile)   
    data.append(next(reader))   #Get header
    for row in reader:          #Iterate each line
        temp = []               #Temp list
        for i in row[-1].strip().split():
            temp.append(row[:-1] + [i])
        data.extend(temp)

with open(filename2, "w", newline='') as outfile:   #Write file
    writer = csv.writer(outfile)
    writer.writerows(data)