Quick Subsetting of Python Array

Question

Coming from R to Python and I see why so many people love Python for data science. One useful feature of R is the quick subsetting. For example:

my_data = c(11,34,67,134,45,8,99,3543,1)
my_subset = c(3,8,1,6,4)
print(my_data[my_subset])
[1]  67  3543  11  8  134

One can programmatically generate subsets that satisfy various conditions and filter the data to that subset with a single instruction. How does one accomplish this in python?

If you have NumPy array, arr = np.array([11,34,67,134,45,8,99,3543,1]); idx = np.array([3,8,1,6,4]); arr[idx-1] should do it, Output -> array([ 67, 3543, 11, 8, 134]) — Ch3steR
– Ch3steR, Commented Feb 27, 2021 at 4:21

Razzle Shazl · Accepted Answer · 2021-02-27 04:27:31Z

1

I'll go ahead and point out the obvious one-liner:

data = (11,34,67,134,45,8,99,3543,1)
indices = (3,8,1,6,4)
subset = [data[i] for i in indices]
print(subset)

Output:

[134, 1, 34, 99, 45]

answered Feb 27, 2021 at 4:27

Razzle Shazl

1,3501 gold badge10 silver badges21 bronze badges

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

Collectives™ on Stack Overflow

Quick Subsetting of Python Array

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related