From b0efd29120a312c0ab32f96bd5a92b8c7c150412 Mon Sep 17 00:00:00 2001 From: Hykilpikonna Date: Sat, 18 Sep 2021 13:50:13 -0400 Subject: [PATCH] [+] A1 p5 --- assignments/a1/a1_part5.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/assignments/a1/a1_part5.py b/assignments/a1/a1_part5.py index 241bbcc..afa316b 100755 --- a/assignments/a1/a1_part5.py +++ b/assignments/a1/a1_part5.py @@ -87,6 +87,9 @@ def separate_colour_channels(rgb_pixels: list) -> dict: 'b': [128, 65, 68, 43, 42, 45, 82, 92, 23, 23, 19, 29, 128, 128, 82, 34]} True """ + return {'r': [p[0] for p in rgb_pixels], + 'g': [p[1] for p in rgb_pixels], + 'b': [p[2] for p in rgb_pixels]} def calculate_median_colour(colour_channels: dict) -> tuple: @@ -102,6 +105,7 @@ def calculate_median_colour(colour_channels: dict) -> tuple: >>> calculate_median_colour(separated_colour_channels) (38, 71, 55) """ + return tuple([int(median(colour_channels[key])) for key in 'rgb']) def apply_median_filter(pixel_data: list, image_width: int, image_height: int) -> list: