Find and remove columns from array based on another array python

Question

I have two .CSV files:

One is a dataset with over 1500 features and 300 samples,

the second is an RFECV ranking of features:

I'm trying to remove each column of a feature from the dataset, that does not have a ranking of 1.

So we only should have something like this:

What would be the proper way of doing something like that in Python?

I was thinking of transposing the second array, finding the indexes with ones and moving columns with these indexes from the dataset to an another array.

Nuri Taş · Accepted Answer · 2022-08-25 13:59:46Z

2

Try:

rank_1 = df2[df2.Ranking == 1].Features
new_df = df1[rank_1]

answered Aug 25, 2022 at 13:59

Nuri Taş

3,8552 gold badges8 silver badges22 bronze badges

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

1 Comment

melson Over a year ago

Thank you very much, It's exactly how I needed it and way simpler too!

Maykon Meneghel · Accepted Answer · 2022-08-25 14:14:12Z

1

import pandas as pd

df1 = pd.read_csv("path-to-first-csv-file.csv")
df2 = pd.read_csv("path-to-second-csv-file.csv")

result = df1[df2[df2["Ranking"] == 1]["Features"]]

answered Aug 25, 2022 at 14:14

Maykon Meneghel

3857 silver badges10 bronze badges

Collectives™ on Stack Overflow

Find and remove columns from array based on another array python

2 Answers 2

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related