How to update a 2d numpy array?

Question

I have the following array:

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

What is the best way to parse each of its elements and assign a string depending on the value? For example: if value < 0 then set "LOW" else set "HIGH"?

You can't assign a string to a numpy array that already contains numbers. Do you want to create a new array or, list-of-lists with the results? — Hannes Ovrén
– Hannes Ovrén, Commented Feb 9, 2015 at 7:49

plonser · Accepted Answer · 2015-02-09 08:11:00Z

1

If a is your array use

import numpy as np

np.where(a<0,'LOW','HIGH')

Edit: When you have 3 choices you can do something like

b = np.where(a < 0.,'LOW','HIGH').astype('S7') 

c = np.where((-1. < a) & (a < 1.), 'MIDDLE',b)

edited Feb 9, 2015 at 8:11

answered Feb 9, 2015 at 7:51

plonser

3,3632 gold badges21 silver badges23 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How to update a 2d numpy array?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related