Ad
Numpy Function That Rounds To Specified Integer To Specified Range
Lets say I have an array y which contains continuos numbers which mostly go from -1 to 5 with probably some outliers but I want to fix them to specified integers(0,1,2,3,4) by giving them specific points where the ranges of integers end ie. [0.7,1.6,2.4,3.7] so here 0 would be everything below 0.7, 1 would be everything between 0.7 and 1.6, 2 would be between 1.6 and 2.4 etc. I'm wondering if there's a function in numpy that can do this for me more efficiently than the code below. I've read the docs of numpy.fix and numpy.rint but I dont see how i can do that with them.
Here's an example of what I want to do basically:
def flatten(y):
for i in range(len(y)):
if y[i] <0.7:
y[i] = 0
elif y[i]>0.7 and y[i]<1.6:
y[i] = 1
elif y[i]>1.6 and y[i]< 2.4:
y[i] = 2
elif y[i] >2.4 and y[i]<3.7:
y[i] = 3
elif y[i]> 3.7:
y[i] = 4
return y
Doesn't have to be a single function but at least something more efficient than this.
Ad
Answer
You can use np.digitize
:
np.digitize(y, [0.7, 1.6, 2.4, 3.7])
Ad
source: stackoverflow.com
Related Questions
- → What are the pluses/minuses of different ways to configure GPIOs on the Beaglebone Black?
- → Django, code inside <script> tag doesn't work in a template
- → React - Django webpack config with dynamic 'output'
- → GAE Python app - Does URL matter for SEO?
- → Put a Rendered Django Template in Json along with some other items
- → session disappears when request is sent from fetch
- → Python Shopify API output formatted datetime string in django template
- → Can't turn off Javascript using Selenium
- → WebDriver click() vs JavaScript click()
- → Shopify app: adding a new shipping address via webhook
- → Shopify + Python library: how to create new shipping address
- → shopify python api: how do add new assets to published theme?
- → Access 'HTTP_X_SHOPIFY_SHOP_API_CALL_LIMIT' with Python Shopify Module
Ad