Question

我有一个包含多个文件的文件夹，每个文件在每个文件中都有不同数量的列。我想浏览目录，打开每个文件并遍历每一行，根据该行中的列数将行写入新的CSV文件。我想最终得到一个包含14列的所有行的大CSV，包含18列的所有行的另一个大CSV，以及包含所有其他列的最后一个CSV。

这是我到目前为止所拥有的。

import pandas as pd
import glob
import os
import csv


path = r'C:\Users\Vladimir\Documents\projects\ETLassig\W3SVC2'
all_files = glob.glob(os.path.join(path, "*.log")) 

for file in all_files:
    for line in file:
        if len(line.split()) == 14:
            with open('c14.csv', 'wb') as csvfile:
                csvwriter = csv.writer(csvfile, delimiter=' ')
                csvwriter.writerow([line])
        elif len(line.split()) == 18:
            with open('c14.csv', 'wb') as csvfile:
                csvwriter = csv.writer(csvfile, delimiter=' ')
                csvwriter.writerow([line])          
            #open 18.csv
        else:
            with open('misc.csv', 'wb') as csvfile:
                csvwriter = csv.writer(csvfile, delimiter=' ')
                csvwriter.writerow([line])
print(c14.csv)

任何人都可以提供有关如何处理此问题的任何反馈吗？

Answer 1

您可以将所有列添加为列表中的列表：

l = []
for file in [your_files]:
    with open(file, 'r') as f:
        for line in f.readlines()
            l.appned(line.split(" "))

现在你有了一个列表列表，所以只需用子列表的长度对它们进行排序，然后将它放在一个新文件中：

l.sort(key=len)

with open(outputfile, 'w'):
     # Write  lines here as you want

Answer 2

请事先注意，您可以将原样行从输入文件中复制到输出文件中，而无需CSV机制。

也就是说，我建议使用文件对象字典和get字典方法，该方法允许指定默认值。

files = {14:open('14.csv', 'wb'),
         18:open('18.csv', 'wb')}
other = open('other.csv', 'wb')

for file in all_files:
    for line in open(file):
        llen = len(line.split())
        target = files.get(llen, other)
        target.write(line)

如果您必须处理几百万条记录，请注意，因为

In [20]: a = 'a '*20                                                                      

In [21]: %timeit len(a.split())                                                           
599 ns ± 1.59 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

In [22]: %timeit a.count(' ')+1                                                           
328 ns ± 1.28 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

您应将上面的for循环替换为

for file in all_files:
    for line in open(file):
        fields_count = line.count(' ')+1
        target = files.get(fields_count, other)
        target.write(line)

应该是因为，即使我们说的是纳秒，文件系统访问也位于同一场所。

In [23]: f = open('dele000', 'w')                                                         

In [24]: %timeit f.write(a)                                                               
508 ns ± 154 ns per loop (mean ± std. dev. of 7 runs, 1000000 loops each)

作为拆分/计数。

Python - 根据列从多个文件中排序csv数据

2 个答案: