Vinícius Aguiar - 9 months ago 51

Python Question

I need to have a list containing all specific element's columns for every index. For example, this DataFrame:

`>>> df`

1 2 3 4 5

2016-01-27 A B B I I

2016-03-07 A C D U U

2016-04-12 H A V V V

2016-05-02 B L Y S N

2016-05-23 L N N A S

Inputting "A" I'd like to have this list as output:

`[1,1,2,NaN,4]`

Is there a built-in method for this?

Answer Source

You may want to `melt`

your data frame to long format and then calculate the corresponding list of columns for each input(value), After obtaining the Series as follows, it would be easy for you to query the result for any intended input:

```
import pandas as pd
pd.melt(df).groupby('value').variable.apply(list)
#value
#A [1, 1, 2, 4]
#B [1, 2, 3]
#C [2]
#D [3]
#H [1]
#I [4, 5]
#L [1, 2]
#N [2, 3, 5]
#S [4, 5]
#U [4, 5]
#V [3, 4, 5]
#Y [3]
#Name: variable, dtype: object
```

To get the list of columns for input `A`

:

```
result = pd.melt(df).groupby('value').variable.apply(list)
result['A']
# ['1', '1', '2', '4']
```