Return column name from index and value with pandas

Question

I am looking for a way to find a column name with index number and value.

original = {'col1': [1, 2,0,9], 'col2': [3, 4, 2,5]}
original = pd.DataFrame(data=original)

    col1    col2
0   1        3
1   2        4
2   0        2
3   9        5

For example, I want to find where is 2 is recorded in the column. I only know index number 1 and value 2 but do not know what is column name.

I am expecting to return the only column name col1.

cs95 · Accepted Answer · 2019-05-25 04:04:27Z

1

Use loc to select the row, and idxmax to select the column:

(original.loc[1] == 2).idxmax()
# 'col1'

If "1" is an integer position rather than index label, use iloc for row selection:

(original.loc[1] == 2).idxmax()
# 'col1'

answered May 25, 2019 at 4:04

cs95

406k106 gold badges744 silver badges797 bronze badges

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

Comments

Rajat Jain · Accepted Answer · 2019-05-25 04:28:58Z

0

This solutions uses numpy.argwhere:

original.columns[np.argwhere(original.iloc[1,:] == 2)[0][0]]

answered May 25, 2019 at 4:28

Rajat Jain

2,0422 gold badges17 silver badges23 bronze badges

Collectives™ on Stack Overflow

Return column name from index and value with pandas

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related