Lista w pythonie

Question

Napisałem funkcje, która zwraca listę plików z katalogu i pod katalogów w takiej formie:

['plik1.txt']
['plik2.txt']
['plik3.txt']

a chciałbym otrzymać taki zapis: ['plik1.txt','plik2.txt','plik3.txt']. Jak osiągnąć taki efekt? Jakaś podpowiedź lub sugestia?

import os
def file(root):
    for path, subdirs, files in os.walk(root):
        for file in files:
            li = []
            li.append(file)
            print(li)
x = 'C:/Users/'

file(str(x))

 

Answer 1

Za każdym razem zerujesz listę li w pętli for. Wypróbuj następującą modyfikację:

import os
def file(root):
    li =[]
    for path, subdirs, files in os.walk(root):
        for file in files:
            li.append(file)
    return li

x = 'C:/Users/'

y = file(str(x))
print y

 

Comments

dzięki o to chodziło