我有一个这样的数组：
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]]
通过使用numpy以下方法可以大大促进（并加快）：
import numpy as np
Y = (np.array(X) == 2).astype(int)
>>> Y
array([[0, 1, 0],
[0, 1, 0],
[1, 0, 0]])
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句