折腾我几天的符号矩阵,原来用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]) #我索你