extracting particular column using header name in python

Question

I am trying to learn python and a beginner. I have a tab delimited text file and I would like to extract 2nd column of the file using the header name of the 2nd column. This is my dummy file:

col1    col2    col3    col4
a   1   2   3
d   4   5   0
f   3   2   1

I have done something like this:

f = open('test.txt',"r")
header = f.readline()
header = firstline.rstrip("\n").split("\t")
for row in f:
 row = row.rstrip("\n")
 # i cannot figure out what should I do next

I want the output to be

col2
1
4
3

rftr · Accepted Answer · 2020-11-19 14:29:07Z

1

You could store the index of your column in a variable:

f = open("test.txt", "r")
col_index = f.readline().rstrip("\n").split("\t").index("col2")
for row in f:
    print(row.rstrip("\n").split("\t")[col_index])

answered Nov 19, 2020 at 14:29

rftr

1,2752 gold badges16 silver badges22 bronze badges

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

Collectives™ on Stack Overflow

extracting particular column using header name in python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related