Splitting a list in Pandas Dataframe to multiple columns

Question

I have a pandas dataframe like shown below where the coordinates column contains X and Y coordinates:

    Coordinates Cluster
0   [25, 79]    2
1   [34, 51]    2
2   [22, 53]    2
3   [27, 78]    2
4   [33, 59]    2

I want to split the Coordinates column into X and Y column so that I have something like below:

    X   Y   Cluster
0   25  79  2
1   34  51  2
2   22  53  2
3   27  78  2
4   33  59  2

How can I achieve this?

BENY · Accepted Answer · 2020-10-02 14:37:50Z

1

Check with

out = df.join(pd.DataFrame(df.pop('Coordinates').tolist(), index=df.index, columns=["X", "Y"]))

edited Oct 2, 2020 at 14:37

answered Oct 2, 2020 at 14:26

BENY

324k22 gold badges176 silver badges250 bronze badges

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

3 Comments

raziiq Over a year ago

Thanks for the answer, it worked. Currently, it changes the name of the columns to 0 and 1, how can create those columns with specific names, e.g., X and Y?

raziiq Over a year ago

I guess, I figured that out myself. Here is what I did out = df.join(pd.DataFrame(df.pop('Coordinates').tolist(), index=df.index, columns=["X", "Y"]))

BENY Over a year ago

@RaziIqbal you got it ~

sammywemmy · Accepted Answer · 2020-10-02 14:45:10Z

0

You could dump it into numpy as well :

df = pd.DataFrame(
{
    "Coordinates": [[25, 79], [34, 51], [22, 53], [27, 78], [33, 59]],
    "Cluster": [2, 2, 2, 2, 2],
})


box = df.to_numpy()

pd.DataFrame(np.column_stack([np.vstack(box[:, 0]), box[:, -1]]),
            columns=["X", "Y", "Cluster"])


    X   Y   Cluster
0   25  79  2
1   34  51  2
2   22  53  2
3   27  78  2
4   33  59  2

answered Oct 2, 2020 at 14:45

sammywemmy

28.9k4 gold badges21 silver badges35 bronze badges

Comments

Mario · Accepted Answer · 2020-10-02 17:15:15Z

0

df = pd.DataFrame(df.Coordinates.str.split(',',1).tolist(),
                                 columns = ['X','Y'])

edited Oct 2, 2020 at 17:15

Mario

2,0844 gold badges34 silver badges81 bronze badges

answered Oct 2, 2020 at 14:33

Mitchell

2371 silver badge8 bronze badges

Collectives™ on Stack Overflow

Splitting a list in Pandas Dataframe to multiple columns

3 Answers 3

3 Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

3 Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related