python - 在Python中根据另一个数组中的索引创建一个数组

Question

我有一个这样的数组：

X= [[1,2,3],
    [3,2,1],
    [2,1,3]]

现在我想创建另一个数组 Y。Y 中的元素应该在 X 中的元素等于 2 的位置取值 1，否则它们应该取值 0。在这个例子中，Y 应该等于

Y=[[0,1,0],
   [0,1,0],
   [1,0,0]]

标签： pythonarrays

Answer 1

使用以下命令将极大地促进（并加快速度）numpy：

import numpy as np

Y = (np.array(X) == 2).astype(int)

>>> Y
array([[0, 1, 0],
       [0, 1, 0],
       [1, 0, 0]])