python的一个for循环用法

本来想对一个数组的每个元素进行求cos。
但是由于我是直接进行矩阵乘法

 x = np.multiply(p , math.cos(w))

报错

only size-1 arrays can be converted to Python scalars

发现是因为math.cos不能直接对矩阵进行运算。
后来尝试寻找如何对矩阵的每一个元素进行操作。
发现一个文章
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]))

对于里面有段代码有点疑惑

for j in range(len(zhen1[i]))] for i in range(len(zhen1))

然后分开逐步尝试

for i in range(len(zhen1)):
    for j in range(len(zhen1[i])):
        print(zhen1[i][j])

第一行的len(zhen1)为2，即2行，i是可以0，1
第二行的len(zhen1[i])为4，即4列，j可以是0，1，2，3
输出顺序是
1（第一行第一列
2（第一行第二列
3（第一行第三列
4（第一行第四列
5（第二行第一列
6。。
7。。
8。。