python - 拆分分隔文件并存储到新列中

Question

我正在尝试拆分 csv 文件。阅读分隔文件后，我想进一步拆分所需的列。我的示例代码:

import csv
sample = open('~/sample.txt')
adr = csv.reader(sample, delimiter='|')
for row in adr:
    a = row[0]
    b = row[1]
    c = row[2]
    d = row [3]
    new=""
    new = row[4].split(",")
    for row1 in new:
        print row1

sample.txt 文件包含:

aa|bb|cc|dd|1,2,3,4|xx
ab|ax|am|ef|1,5,6|jk
cx|kd|rd|j|1,9|k

以上代码产生的输出为:

[1,2,3,4]
[1,5,6]
[1,9]

我正在尝试进一步拆分新列并使用拆分输出进行比较。例如，所需的拆分输出为:

aa|bb|cc|dd|1|2|3|4|xx
ab|ax|am|ef|1|5|6| |jk
cx|kd|rd|j|1|9| | |k

我还想存储新列的多个空白或NULL值，如上面的示例[1,2,3,4]，[1,5,6]所示。有更好的分割方式吗？

Answer 1

你已经差不多了!您只需要在 new = row[4].split(",") 之后再添加几行即可。

for i in range(len(new), 4):
    new.append('')
newrow = row[0:4] + new + row[5:]
print('|'.join(newrow))

编辑2:以最简单的方式处理下面的评论，只需循环两次，第一次寻找最长的“子数组”。回复:打印额外次数，您可能将代码复制到错误的位置/缩进并将其放入循环中。

完整代码:

import csv
sample = open('~/sample.txt')
adr = csv.reader(sample, delimiter='|')
longest = 0
for row in adr:
    curLen = len(row[4].split(','))
    if curLen > longest:
        longest = curLen

sample.seek(0)
for row in adr:
    new = row[4].split(",")
    for i in range(len(new), longest):
        new.append(' ')
    newrow = row[0:4] + new + row[5:]
    print('|'.join(newrow))