Squaring all elements in a list

Question

I am told to

Write a function, square(a), that takes an array, a, of numbers and returns an array containing each of the values of a squared.

At first, I ha

Answer 1

from array import *

array('i', [5, 6, 4])
print(*list(map(pow, vals, [2] * len(vals))), sep='\n')