Python Forum
How to find column index and its corresponding column name
Thread Rating:
  • 0 Vote(s) - 0 Average
  • 1
  • 2
  • 3
  • 4
  • 5
How to find column index and its corresponding column name
#2
The problem I see is that your paraList is not really a data frame as it has only one row... so looks like it is going to work better as a dictionary:
datapara.iloc[0].to_dict()
Output:
{'A': 'Y', 'C': 'N', 'E': 'Y', 'F': 'Y'}
Now it is easy to create a selector for the columns that has value 'Y'
selection = datapara.iloc[0].to_dict()
cols = [c for c in selection if selection[c] == 'Y']
print(dataRaw[cols])
Output:
A E F 0 0 3.2 1.6 1 1 3.2 1.6 2 2 6.5 1.9 3 0 3.2 1.6 4 1 6.5 1.9 5 4 3.2 1.6
To select the ones with 'N', you can use a similar process.
Remember also to add guards for the cases when a column is only in one of the tables... for example in this code a column only in datapara will raise an exception.
Reply


Messages In This Thread
RE: How to find column index and its corresponding column name - by killerrex - May-09-2018, 09:53 PM

Possibly Related Threads…
Thread Author Replies Views Last Post
  Column Transformer with Mixed Types - sklearn aaldb 0 409 Feb-22-2024, 03:27 PM
Last Post: aaldb
  concat 3 columns of dataframe to one column flash77 2 932 Oct-03-2023, 09:29 PM
Last Post: flash77
  HTML Decoder pandas dataframe column mbrown009 3 1,143 Sep-29-2023, 05:56 PM
Last Post: deanhystad
  attempt to split values from within a dataframe column mbrown009 8 2,531 Apr-10-2023, 02:06 AM
Last Post: mbrown009
  Finding the median of a column in a huge CSV file markagregory 5 1,892 Jan-24-2023, 04:22 PM
Last Post: DeaD_EyE
  Make unique id in vectorized way based on text data column with similarity scoring ill8 0 928 Dec-12-2022, 03:22 AM
Last Post: ill8
  Impute 1 if previous row of 'days' column is between 0 & 7 JaneTan 2 1,170 Dec-08-2022, 07:42 PM
Last Post: deanhystad
  Increase df column values decimals SriRajesh 2 1,153 Nov-14-2022, 05:20 PM
Last Post: deanhystad
  pandas column percentile nuncio 7 2,555 Aug-10-2022, 04:41 AM
Last Post: nuncio
  how to expand each unique value in another column and fill zero if no match SriRajesh 0 869 Jul-10-2022, 09:21 AM
Last Post: SriRajesh

Forum Jump:

User Panel Messages

Announcements
Announcement #1 8/1/2020
Announcement #2 8/2/2020
Announcement #3 8/6/2020