在条件下合并一些行

Question

我想合并条件中的行。如果该行的长度小于 20，则将该行与之前的行合并。 我正在尝试使用此代码，但输出不正确 感谢您的帮助。

import csv 

datam = []
with open("All.csv", 'r', encoding='UTF-8') as f:
  lines = f.readlines()
for line in lines:
    doc = line
    datam.append(doc)
with open('Test.csv', mode = 'w',newline ='',encoding='UTF-8') as f:
    writer = csv.writer(f,delimiter=' ')
    for i, data in enumerate(datam):
        if len(datam)< 20:
            writer.writerow([datam[i-1] + ',' + datam[i]])
            datam.remove(data[i])
        if len(data)>= 20:
            writer.writerow([datam[i]])

Answer 1

这里有一些问题。

您的代码似乎正在检查datum 的长度——这是行数，而不是特定行的长度。

我们可以通过在每次迭代中始终检查 next 行的长度来重新安排逻辑。如果它小于 20，我们将它与当前行合并并写入输出 CSV；然后删除该行。否则，我们只需将当前行写入 CSV。

有一个边缘情况，我们在输入 CSV 的最后一行，在这种情况下，我们没有后续行要检查，所以我们直接将该行写入输出 CSV。

最后，开头的代码可以大大简化。这有点令人困惑，因为您将源文件读入lines，然后将其逐行复制到datam。你可以做到这一点。

此外，还有一个小错误，即原始文件将包含换行符，您需要删除这些换行符。

以下内容应该适合您：

import csv 

with open('All.csv', 'r', encoding='utf-8') as f:
  datam = [l.strip() for l in f.readlines()]

with open('Test.csv', mode='w', newline='', encoding='utf-8') as f:
    writer = csv.writer(f, delimiter=' ')
    rows = []
    for i, data in enumerate(datam):
        if i + 1 == len(datam):
            writer.writerow([data])
        elif len(datam[i + 1]) < 20:
            writer.writerow([data + ',' + datam[i + 1]])
            datam.remove(datam[i + 1])
        elif len(datam[i + 1]) >= 20:
            writer.writerow([data])