我有以下数组:
array([[ 0. , -1.22474487, 1.40182605],
[ 1.22474487, 0. , -0.53916387],
[-1.22474487, 1.22474487, -0.86266219]])
解析每个元素并根据值分配字符串的最佳方法是什么?例如:if value< 0然后设置" LOW"否则设置" HIGH"?
答案 0 :(得分:1)
如果您a
使用array
import numpy as np
np.where(a<0,'LOW','HIGH')
编辑: 如果您有3个选择,则可以执行类似
的操作b = np.where(a < 0.,'LOW','HIGH').astype('S7')
c = np.where((-1. < a) & (a < 1.), 'MIDDLE',b)