python中numpy实现符号矩阵&布尔索引

折腾我几天的符号矩阵，原来用numpy的where函数即可。

将一个随机生成的矩阵数据，转换成符号矩阵，其实就是将其中的正值都替换成1，将所有的负值替换为-1，使用np.where即可实现

import numpy as np
x = np.random.randn(3,3)
#x的符号矩阵y
y = np.where(x>0,1,-1)

笨一点的方法就是

import numpy as np
z = np.random.randn(3,3)
z_sign = z.copy()
z_sign[z_sign > 0] = 1
z_sign[z_sign < 0] = -1
print(z_sign)

补充一个
布尔索引
可以用在两个关联数组上获取指定数据

import numpy as np
x = np.array(['A','B','C','D','E'])
y = np.array([1,2,3,4,5])
print(y[x == 'A']) #你索我
print(x[y == 1]) #我索你