CSV解析和转换答案

【问题标题】：CSV parsing and conversionCSV解析和转换
【发布时间】：2014-10-05 15:21:22
【问题描述】：

我正在寻找按如下方式组织的 csv 文件：

<data1>,<data2>
asdf,<data3>
asdf,<data4>
asdf,<data5>
<data6>,<data7>
asdf,<data8>

<data1>,<data2>
asdf,<data3>
asdf,<data4>
asdf,<data5>
<data6>,<data7>
asdf,<data8>

<data1>,<data2>
asdf,<data3>
asdf,<data4>
asdf,<data5>
<data6>,<data7>
asdf,<data8>

etc.

我正在尝试输出一个看起来像这样的 .csv：

<data1>,<data2>,<data3>,<data4>,<data6>,<data7>,<data8>
<data1>,<data2>,<data3>,<data4>,<data6>,<data7>,<data8>
etc.

有人可以帮我解决这个问题吗？

编辑：想通了，如果有人感兴趣的话..

import csv

with open('C:\Temp\eqtest.csv', 'rb') as inf, open('C:\Temp\output.csv', 'wb') as outf:
    reader = csv.reader(inf)
    writer = csv.writer(outf)
    i = -1
    line = []
    for row in reader:
        print(line)
        print(i)
        print(row)
        while row == ['','']:
            row = next(reader)
        i += 1
        if i == 0 or i == 4:
            line.append(row[0])
            line.append(row[1])
        elif i == 2 or i == 3:
            line.append(row[1])
        elif i == 5:
            line.append(row[1])
            i = -1
            writer.writerow(line)
            line = []

【问题讨论】：

您是否尝试过自己编写代码？如果是这样，请显示您的代码以及您的错误是什么。
大概中间没有真正的空行？
@MartijnPieters 有空行
@andrejr：有多少？任意数字？
@MartijnPieters 是的。只要输入中有换行符，程序就应该输出一个新行（无论中间有多少空行）......

标签： python parsing csv

【解决方案1】：

您可以将csv.reader() 用作可迭代对象，并使用next() 或itertools.islice() 获取更多行：

import csv
from itertools import islice

with open('input.csv', 'rb') as inf, open('output.csv', 'wb') as outf:
    reader = csv.reader(inf)
    writer = csv.writer(outf)
    for row in reader:
        while not row:
            # skip empty rows
            continue

        result = row
        for extra_row in islice(reader, 3):
            result.append(extra_row[1])
        result.extend(next(reader))
        result.append(next(reader)[1])

        writer.writerow(result)

这会从读取器中取出一行，并使用所有列作为输出行的开始。然后它从同一个 CSV 中再提取 3 行来获取第二列，并将其添加到输出行。使用next() 会额外读取两行，将整行和一列添加到输出中。

跳过每个 6 行块之前的任何空行。

然后读取输出，for 循环的下一次迭代可以开始，此时已经读取了 6 个实际行，并且循环从输入文件的第 7 行开始；如果这是空的，阅读器会前进，直到找到一个非空行。

演示：

>>> import csv
>>> import sys
>>> from itertools import islice
>>> sample = '''\
... <data1>,<data2>
... asdf,<data3>
... asdf,<data4>
... asdf,<data5>
... <data6>,<data7>
... asdf,<data8>
... 
... <data1>,<data2>
... asdf,<data3>
... asdf,<data4>
... asdf,<data5>
... <data6>,<data7>
... asdf,<data8>
... 
... <data1>,<data2>
... asdf,<data3>
... asdf,<data4>
... asdf,<data5>
... <data6>,<data7>
... asdf,<data8>
... '''.splitlines()
>>> reader = csv.reader(sample)
>>> writer = csv.writer(sys.stdout)
>>> for row in reader:
...     while not row:
...         # skip empty rows
...         continue
...     result = row
...     for extra_row in islice(reader, 3):
...         result.append(extra_row[1])
...     result.extend(next(reader))
...     result.append(next(reader)[1])
...     writer.writerow(result)
... 
<data1>,<data2>,<data3>,<data4>,<data5>,<data6>,<data7>,<data8>
<data1>,<data2>,<data3>,<data4>,<data5>,<data6>,<data7>,<data8>
<data1>,<data2>,<data3>,<data4>,<data5>,<data6>,<data7>,<data8>

【讨论】：