python number of elements in multidimensional array

#Example for a 2D list
an_array = [[1, 2], [3, 4, 5]]
total_length = sum(len(row) for row in an_array)
print(total_length)
#OUTPUT: 5

Here is what the above code is Doing:
1. We create a 2D list called an_array.
2. We use a list comprehension to iterate through each row in an_array.
3. We use the built-in len() function to get the length of each row.
4. We use the built-in sum() function to sum up the lengths of each row.
5. We print the total length of all the rows in an_array.