Square with map

Author: Richouf95

0x04-python-more_data_structures/100-main.py

@@ -0,0 +1,13 @@
+#!/usr/bin/python3
+square_matrix_map = \
+    __import__('101-square_matrix_map').square_matrix_map
+
+matrix = [
+    [1, 2, 3],
+    [4, 5, 6],
+    [7, 8, 9]
+]
+
+new_matrix = square_matrix_map(matrix)
+print(new_matrix)
+print(matrix)

0x04-python-more_data_structures/101-main.py

@@ -0,0 +1,3 @@
+#!/usr/bin/python3
+def square_matrix_map(matrix=[]):
+    return (list(map(lambda row: list(map(lambda x: x**2, row)), matrix)))