python输入数组并平方

Question

用 input 输入数组 还要用到 x.astype(int) ，最终输出数组的平方（要求是个数组）

Answer 1

numpy是支持多维数组平方的，没其它人那么复杂

import numpy as np
x_in = np.array(eval(input('输入数组'))).astype(int)
x_out = x_in**2
print(x_out)

输入数组1,2,3,4
[ 1  4  9 16]