diff --git a/19-defensive/tf-19.py b/19-defensive/tf-19.py index 7dea0bd..578dce5 100644 --- a/19-defensive/tf-19.py +++ b/19-defensive/tf-19.py @@ -64,7 +64,11 @@ def sort(word_freq): # # The main function # -word_freqs = sort(frequencies(extract_words(sys.argv[1]))) +#if (len(sys.argv) < 2): +# filename = "../input.txt" +#else: +filename = sys.argv[1] if len(sys.argv) > 1 else "../input.txt" +word_freqs = sort(frequencies(extract_words(filename))) for tf in word_freqs[0:25]: print tf[0], ' - ', tf[1]