python - 无法重写代码以列出理解python中图像旋转的代码

Question

因此，在stackoverflow上找不到我的问题之后，这是如何将for循环重写为列表推导式(在其中插入值的地方)，我现在必须询问是否可以重写此代码:

rotate = np.zeros((w,h,c), np.uint8) # create an empty image filled with zeros turned 90°

for y in xrange(h):
    for x in xrange(w):
        rotate[x][y] = img[y][x]

进入列表理解？我以为这样的事情会起作用，但是没有:

rotate[x][y] = img[y][x] for y in range(h) for x in range(w) 

在那之后，我只是尝试添加索引和方括号的各种组合，并且总是遇到一些语法错误。仅作记录，我知道在opencv和numpy中都有用于旋转图像的函数，我只是想将for循环重写为列表理解。

Answer 1

rotate = np.array([[img[y][x] for y in xrange(h)] for x in xrange(w)])