Python: using where condition [duplicate]

Question

Is it possible to compute the following using single where condition in Python. I'm unable to do that.

x = np.arange(10)

If an element of x is smaller than 3, replace it with 3. And if an element of x is bigger than 7, replace it with 7.

My attempt is as follows, however I'm wondering if it could be done in a single line of code

x= np.where(x<3 ,3,x) x = np.where(x>7,7,x)

Sorry, if it seems very basic. But I have just started with numpy

Ali Hashemi · Accepted Answer · 2017-06-24 07:55:41Z

1

You can do it in a single line using numpy.clip()

x = np.arange(10)
np.clip(x, 3, 7)
print(x)

#[3, 3, 3, 3, 4, 5, 6, 7, 7, 7]

answered Jun 24, 2017 at 7:55

Ali Hashemi

3,4283 gold badges37 silver badges48 bronze badges

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

Comments

danche · Accepted Answer · 2017-06-24 07:53:10Z

0

In [115]: a = np.arange(10)

In [116]: np.clip(a, 3, 7)
Out[116]: array([3, 3, 3, 3, 4, 5, 6, 7, 7, 7])

answered Jun 24, 2017 at 7:53

danche

1,81517 silver badges23 bronze badges

vZ10 · Accepted Answer · 2017-06-24 07:58:05Z

-1

x= np.where(x<3, 3, np.where(x>7, 7, x))

answered Jun 24, 2017 at 7:58

vZ10

2,7662 gold badges25 silver badges34 bronze badges