How to select ndarray data in python?

Question

I have a ndarray type array, for example:

1, 2, 0.5
2, 6, 0.9
9, 2, 0.83

I want to keep the rows whose 3rd elements are larger than 0.8, and discard the other rows. It means I want this result:

2, 6, 0.9
9, 2, 0.83

How can I finish the task?

Sumit S Chawla · Accepted Answer · 2017-12-11 12:37:54Z

2

Here is a simple implementation of your problem:

import numpy as np
data = np.array([[1, 2, 0.5],[2, 6, 0.9],[9, 2, 0.83]])
result =data[data[:,2]>0.8]

Output:

[[ 2.    6.    0.9 ]
[ 9.    2.    0.83]]

answered Dec 11, 2017 at 12:37

Sumit S Chawla

3,6701 gold badge16 silver badges34 bronze badges

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

1 Answer 1