python - 在花式索引时使用范围？

Question

有人能用简单的术语向我解释一下这个表达式吗？您如何获取两个值的范围并将它们设置为零？

lena_image[range(xmax), range(ymax)] = 0

Answer 1

您的代码会将主对角线设置为零:

import numpy as np

n = 5
lena_image = np.ones(shape=(n, n))
lena_image[range(n), range(n)] = 0
print lena_image

输出:

[[ 0.  1.  1.  1.  1.]
 [ 1.  0.  1.  1.  1.]
 [ 1.  1.  0.  1.  1.]
 [ 1.  1.  1.  0.  1.]
 [ 1.  1.  1.  1.  0.]]