python中对二维数组每个元素进行相同操作，以平方为例

import numpy as np
from numpy.matlib import repmat
zhen1=np.array([[1,2,3,4],[5,6,7,8]])
#zhen=np.array([[1,1,1,1]])

zhen2= [[zhen1[i][j]**2 for j in range(len(zhen1[i]))] for i in range(len(zhen1))]
print(zhen2,np.shape(zhen2),type(zhen2),type(zhen2[0]),type(zhen2[0][0]))import numpy as np
from numpy.matlib import rep