Extracting All Second Column Values From A Ndarray For A Given First Column Value
I have a numpy ndarray where first column is user id and second column is some product id. What would be the fastest way to get all product ids for a given user id?
I've been going through the numpy doc and this handbook (https://jakevdp.github.io/PythonDataScienceHandbook/02.02-the-basics-of-numpy-arrays.html) as well but I had no luck.
Say we have this array:
test = [[0, 1], [0, 20], [0, 30], [1, 11], [1, 23], [1, 45]]
My goal is to get a function like this:
get_product_ids(0) >> [1, 20, 30]
This can be achieved in such a simple way
test = np.array(test) def get_product_id(ind): mask = test[:, 0] == ind return test[:, 1][mask]
- → 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
- → 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