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]))
输出结果:
E:\phthon35\python.exe I:/catsVSdogs1-master/catsVSdogs1-master/file01/test15.py
[[1, 4, 9, 16], [25, 36, 49, 64]] (2, 4) <class 'list'> <class 'list'> <class 'numpy.int32'>
Process finished with exit code 0