Question

我想拆分python字典并根据NO_OF_LINES_PER_FILE和字典大小将其写入不同的文件

输入

NO_OF_LINES_PER_FILE

所以如果NO_OF_LINES_PER_FILE = 2且字典大小为10，我希望将字典拆分为5个文件（每个文件将有2行）

脚本

import csv
NO_OF_LINES_PER_FILE = 2
s = {"2222":["1","2","3"],"3456":["2","3","4"],"5634":["4","5"],"23543":["456","3"],"29587":["4","5"],"244":["23","34"],"455":["3","4"],"244221":["5"],"23232345":["2323","43"]}

def again(c,h,NO_OF_LINES_PER_FILE1):
    f3 = open('num_'+str(h)+'.csv', 'at')
    if c == 1:
        ceh = 2
    else:
        ceh = c
    print ceh
    v = 0
    for w in s:
        v = v + 1
        if v < ceh:
            pass
        elif v > NO_OF_LINES_PER_FILE1:
            print "yes"
            NO_OF_LINES_PER_FILE1 = NO_OF_LINES_PER_FILE1 + 1
            h = NO_OF_LINES_PER_FILE1 + 1
            again(c,h,NO_OF_LINES_PER_FILE1)
        else:
            writer = csv.writer(f3,delimiter = ',', lineterminator='\n',quoting=csv.QUOTE_ALL)
            writer.writerow(s[w])

        c = c + 1

def split():
    f3 = open('has_'+str(NO_OF_LINES_PER_FILE)+'.csv', 'at')
    writer = csv.writer(f3,delimiter = ',', lineterminator='\n',quoting=csv.QUOTE_ALL)
    c = 0
    for w in s:

        if c >= NO_OF_LINES_PER_FILE:

            NO_OF_LINES_PER_FILE1 = NO_OF_LINES_PER_FILE + 1
            h = NO_OF_LINES_PER_FILE
            again(c,h,NO_OF_LINES_PER_FILE1)
            break
        else:
            #print NO_OF_LINES_PER_FILE
            writer = csv.writer(f3,delimiter = ',', lineterminator='\n',quoting=csv.QUOTE_ALL)

            writer.writerow(s[w])

        c = c + 1

split()

但是这个脚本不起作用并创建了许多文件

在上面的脚本中NO_OF_LINES_PER_FILE = 2且字典s的大小为9

所以我想要5个文件，前四个文件每个包含2行，第五个文件将包含1行

我该如何解决这个问题？

Answer 1

我的方法首先将dict弄平，然后将平面dict拆分为你想要的长度的子列表

    import csv
    flatDict = [ i  for i in s.items()]
    splitFlatDict = [flatDict[i:i+NO_OF_LINES_PER_FILE] for i in xrange(0,len(flatDict),NO_OF_LINES_PER_FILE)]
    for i,rows in enumerate(splitFlatDict):
      with open(str(i) + '.csv','wb') as f:
        writer = csv.writer(f)
        writer.writerows(rows)

拆分字典并将其写入python

1 个答案: