小编Sch*_*ind的帖子

我通过使用两个for循环创建了一个具有对角线边缘的数组，但是我想知道是否有一种更简单的方法，例如列表理解：

im_diag = np.zeros((im_size, im_size), dtype=np.int8)

for x in range(im_size):
    for y in range(im_size):
        if x+y >= im_size:
            im_diag[x,y] = 1