Answers #141 - Now includes weights as an optional input

UCL-RITS · Nov 27, 2020 · 51072b3 · 51072b3
1 parent de72bd1
commit 51072b3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/week06/average-squares-example/average_squares/squares.py b/week06/average-squares-example/average_squares/squares.py
@@ -51,10 +51,11 @@ def convert_numbers(list_of_strings):
 
 
 if __name__ == "__main__":
-    parser = ArgumentParser(description="Mean of squared numbers")
+    parser = ArgumentParser(description="Weighted mean of squared numbers")
     parser.add_argument("numbers", nargs='+', type=float, help="the list of numbers")
+    parser.add_argument("--weights", "-w", nargs='+', type=float, help="the list of weights")
     arguments = parser.parse_args()
 
-    result = average_of_squares(arguments.numbers)
+    result = average_of_squares(arguments.numbers, arguments.weights)
 
     print(result)