Convert directed graph to horizontal data format in pandas python

Question

I just started with python and know almost nothing about pandas.

so now I have a directed graph which look like this:

from ID	to ID
13	22
13	56
14	10
14	15
14	16

now I need to transform it to horizontal data format like this:

from ID	To 0	To 1	To 2
13	22	56	NAN
14	10	15	16

I find something like Pandas Merge duplicate index into single index but it seems it cannot merge rows into different colums.

Will pandas use NAN to fill in the blanks during this process due to the need to add more columns?

anas laaroussi · Accepted Answer · 2022-11-29 00:17:09Z

0

You can do something like this to transform your dataframe, pandas will automatically add NaN values for any number of columns you have with this solution.

df = df.groupby('from')['to'].apply(list)

## create the desired df from the lists
df = pd.DataFrame(df.tolist(),index=df.index).reset_index()


## Rename columns
df.columns = ["from ID"] + [f"To {i}" for i in range(1,len(df.columns))]

edited Nov 29, 2022 at 0:17

answered Nov 29, 2022 at 0:09

anas laaroussi

1264 bronze badges

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

Collectives™ on Stack Overflow

Convert directed graph to horizontal data format in pandas python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related