Как обобщить файловую функцию, чтобы открыть столько текстовых файлов, сколько указано в параметре? - PullRequest
0 голосов
/ 12 ноября 2018

Я хочу обобщить мой следующий код так, чтобы он содержал столько файлов параметров без жесткого кодирования, как f1 и f2.Как я могу это сделать?Вот мой код.

def wordFreq(f1, f2):
    f1 = open(f1, 'r')  
    f2 = open(f2, 'r')
    file_list = [f1, f2]  
    num_files = len(file_list) 
    wordFreq = {} 

    .....
    .....

    return wordFreq

print(wordFreq('file1.txt','file2.txt'))

1 Ответ

0 голосов
/ 12 ноября 2018

Вы хотите * args

def word_freq(*files):
    file_list = [open(file, 'r') for file in files]
    ...
...