Как записать список в файл чанка - PullRequest
Новая
       30

Как записать список в файл чанка

3 голосов
/ 17 июня 2019

У меня есть список списка, который выглядит следующим образом: 

listThing = [['apple','mango','cherry'],
             ['dog','cat','bird'],
             ['rose','jasmine','sunflower']
             ['hospital','house','school']
             ['chair','table','cupboard']
             ['book','pencil','pen']]

Я хочу записать этот список в файлы, где количество файлов является предопределенным значением. Затем количество списков в каждом файле делится на количество всех списков и количество файлов. Так что если: 

number of file = 3
number of list in each file = number of all lists/number of file = 6/3 = 2

Выводы будут выглядеть так:

file1.txt 

apple
mango
cherry
dog
cat
bird

file2.txt 

rose
jasmine
sunflower
hospital
house
school

file3.txt 

chair
table
cupboard
book
pencil
pen

Вот что я пробовал: 

import math

allList = len(listThing)
numFile = 3
listInFile = math.ceil(allList/numFile)

for i in range(listInFile):
    with open('file'+str(i)+'.txt', 'w') as out:
        for n in range(listInFile):
            # I don't know what should I do next

Я не знаю, как решить эту проблему. Я надеюсь, что кто-то может помочь мне решить эту проблему. Спасибо

Ответы [ 2 ]

2 голосов
/ 17 июня 2019 
import math    

list_of_lists = [['apple', 'mango', 'cherry'],
                 ['dog', 'cat', 'bird'],
                 ['rose', 'jasmine', 'sunflower'],
                 ['hospital', 'house', 'school'],
                 ['chair', 'table', 'cupboard'],
                 ['book', 'pencil', 'pen']]

num_files = 3
all_lists = len(list_of_lists)

lists_per_file = math.ceil(all_lists / num_files)

for i in range(1, num_files + 1):
    with open("file{}.txt".format(i), "w") as file:
        lst_idx = (i-1)*lists_per_file
        for lst in list_of_lists[lst_idx:lst_idx+lists_per_file]:
            for word in lst:
                file.write("{}\n".format(word))
1 голос
/ 17 июня 2019

Попробуйте это: 

import math

listThing = [['apple','mango','cherry'],
             ['dog','cat','bird'],
             ['rose','jasmine','sunflower'],
             ['hospital','house','school'],
             ['chair','table','cupboard'],
             ['book','pencil','pen']]

allList = len(listThing)
numFile = 3
listInFile = int(math.ceil(allList/numFile))
currentFileIndex = None

for e, lt in enumerate(listThing):
    fileIndex = 1 + int(math.floor(e / listInFile))
    if currentFileIndex != fileIndex:
        currentFileIndex = fileIndex
        currentFile = open('file%d.txt' % fileIndex, 'wb')
    for entry in lt:
        currentFile.write(entry.encode('utf8'))
        currentFile.write(b'\n')
...