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

2024-10-17 08:31:46 发布

您现在位置:Python中文网/ 问答频道 /正文

我想拆分python字典,并根据每个文件的行数和字典大小将其写入不同的文件

输入

NO_OF_LINES_PER_FILE

因此,如果每个文件没有\u行\u=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()

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

在上面的脚本中,每个文件的行数为2,字典大小为9

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

我怎样才能解决这个问题?你知道吗


Tags: 文件ofcsvno脚本if字典file1
1条回答
网友
1楼 · 发布于 2024-10-17 08:31:46

我的方法是先平铺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)

相关问题 更多 >