Indexing a dataframe column in Pandas

user_19383871 0
Viewed 47

How can I index values in a column in pandas and make it into a new column? This is what I'm trying to do:

Original:

       Data    
0  0010-AAAA    
1  0010-BBBB    
2  0010-CCCC    
3  0011-DDDD    
4  0011-EEEE    

Adding two columns:
       Data    col_2   col_3  
0  0010-AAAA    0010    AAAA
1  0010-BBBB    0010    BBBB
2  0010-CCCC    0010    CCCC
3  0011-DDDD    0011    DDDD
4  0011-EEEE    0011    EEEE
pandas dataframe slice indexing python
3 Answers

Looks like you need a split:

df[['col_2', 'col_3']] = df['Data'].str.split('-', n=1, expand=True)

output:

        Data col_2 col_3
0  0010-AAAA  0010  AAAA
1  0010-BBBB  0010  BBBB
2  0010-CCCC  0010  CCCC
3  0011-DDDD  0011  DDDD
4  0011-EEEE  0011  EEEE

no dash

Then use a regex with str.extract.

In this case: numbers \d+, followed by non numbers \D+:

df[['col_2', 'col_3']] = df['Data'].str.extract(r'(\d+)(\D+)')

output:

       Data col_2 col_3
0  0010AAAA  0010  AAAA
1  0010BBBB  0010  BBBB
2  0010CCCC  0010  CCCC
3  0011DDDD  0011  DDDD
4  0011EEEE  0011  EEEE

or even: r'(\d+)\W*(\D+)' (digits / optional non-alphanum / non-digits) to handle both cases at once:

df[['col_2', 'col_3']] = df['Data'].str.extract(r'(\d+)\W*(\D+)')

example:

        Data col_2 col_3
0  0010-AAAA  0010  AAAA
1   0010BBBB  0010  BBBB
2  0010-CCCC  0010  CCCC
3   0011DDDD  0011  DDDD
4  0011-EEEE  0011  EEEE
user_163434640 
df[['col_2','col_3']]= df['Data'].str.split("-",expand=True)
df

Data    col_2   col_3
0   0010-AAAA   0010    AAAA
1   0010-BBBB   0010    BBBB
2   0010-CCCC   0010    CCCC
3   0011-DDDD   0011    DDDD
4   0011-EEEE   0011    EEEE
user_34947540

All the python string methods will work, including slicing

df["Data"].str[:4]

0    0010
1    0010
2    0010
3    0011
4    0011
user_48842350
Related