如何更新2d numpy数组?

时间:2015-02-09 07:44:02

标签: python numpy

我有以下数组:

array([[ 0.        , -1.22474487,  1.40182605],
       [ 1.22474487,  0.        , -0.53916387],
       [-1.22474487,  1.22474487, -0.86266219]])

解析每个元素并根据值分配字符串的最佳方法是什么?例如:if value< 0然后设置" LOW"否则设置" HIGH"?

1 个答案:

答案 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)