Split a dataframe column's list into two dataframe columns

Question

I'm effectively trying to do a text-to-columns (from MS Excel) action, but in Pandas.

I have a dataframe that contains values like: 1_1, 2_1, 3_1, and I only want to take the values to the right of the underscore. I figured out how to split the string, which gives me a list of the broken up string, but I don't know how to break that out into different dataframe columns.

Here is my code:

import pandas as pd

test = pd.DataFrame(['1_1','2_1','3_1'])
test.columns = ['values']

test = test['values'].str.split('_')

I get something like: [1, 1], [2, 1], [3, 1].

What I'm trying to get is two separate columns:

col1: 1, 2, 3 col2: 1, 1 ,1

Thoughts? Thanks in advance for your help

root · Accepted Answer · 2016-07-19 21:22:15Z

2

Use expand=True when doing the split to get multiple columns:

test['values'].str.split('_', expand=True)

If there's only one underscore, and you only care about the value to the right, you could use:

test['values'].str.split('_').str[1]

answered Jul 19, 2016 at 21:22

root

34.1k6 gold badges77 silver badges89 bronze badges

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

Comments

EoinS · Accepted Answer · 2016-07-19 21:27:27Z

2

You are close:

Instead of just splitting try this:

test2 = pd.DataFrame(test['values'].str.split('_').tolist(), columns = ['c1','c2'])

answered Jul 19, 2016 at 21:27

EoinS

5,4901 gold badge21 silver badges33 bronze badges

Collectives™ on Stack Overflow

Split a dataframe column's list into two dataframe columns

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related