numpy相关操作

1. 判断元素是否为inf

a = np.inf
b = -1*np.inf
print(np.isinf(a),np.isinf(b)) ==> True,True

2. 取出array里所有取值

a = np.unique(b) # b is a numpy array

3. 把array里的所有重复值都合并，且合并后的内容从大到小排序，此外，还需给出合并后元素和原始元素的映射关系

a = np.array([1,3,5,4,2,7,1,3,4])
pre_merge = np.unique(a)
print("before merge",a)
print("after merge",pre_merge)
mapping_dict = []
for merged_model in pre_merge:
    mapping_dict.append(np.where(a==merged_model)[0].tolist())
print("mapping info",mapping_dict)