python - 不使用for循环编辑python二维数组？

Question

所以，我有一个随机生成的给定二维矩阵:

a = np.random.randn(4,4)

给出输出:

array([[-0.11449491, -2.7777728 , -0.19784241,  1.8277976 ],
   [-0.68511473,  0.40855461,  0.06003551, -0.8779363 ],
   [-0.55650378, -0.16377137,  0.10348714, -0.53449633],
   [ 0.48248298, -1.12199767,  0.3541335 ,  0.48729845]])

我想将所有负值更改为 0，将所有正值更改为 1。如果没有 for 循环，我该如何做到这一点？

Answer 1

您可以使用np.where()

import numpy as np

a = np.random.randn(4,4)
a = np.where(a<0, 0, 1)

print(a)

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