拆分字典并将其写入python中的不同csv文件

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 平铺，然后将 flat 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)