How can I conditionally add columns (string) together in pandas Data Frame?

Question

I need to add column A to B. I could achive this by doing this:

import pandas as pd

sample_dict = {
               'A' : ["entry", "", "", "entry"],
               'B' : ["foo", "foo", "", ""]
              }
sample_frame = pd.DataFrame(sample_dict)
sample_frame['A'] = sample_frame['A'] + sample_frame['B']
del sample_frame['B']

Then I end up with:

          A
0  entryfoo
1       foo
2          
3     entry

While the two columns are now one, I can hardly distinguish between the two values. What I want is this:

            A
0  entry, foo
1         foo
2          
3       entry

So I want to put a ", " in between the values, but only if one of the values is not "". Otherwise I end up with this if I just use sample_frame['A'] = sample_frame['A'] + ", " + sample_frame['B']:

            A
0  entry, foo
1       , foo
2           ,      
3      entry,

So, how can I achive this conditional addition?

ThePyGuy · Accepted Answer · 2021-08-13 15:02:08Z

2

Even your solution is fine, just strip off the leftovers at last, using Series.str.strip:

sample_frame['A'] = (sample_frame['A'] + ", " + sample_frame['B']).str.strip(', ')

OUTPUT:

            A
0  entry, foo
1         foo
2            
3       entry

answered Aug 13, 2021 at 15:02

ThePyGuy

18.6k5 gold badges24 silver badges55 bronze badges

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

Collectives™ on Stack Overflow

How can I conditionally add columns (string) together in pandas Data Frame?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related