How to append value to list in a row based on another column python?

Question

I have a dataframe that looks like:

        body              label
   the sky is blue        [noun]
   the apple is red.      [noun]
   Let's take a walk      [verb]

I want to add an item to the list in label depending on if there is a color in the body column.

Desired Output:

            body              label
       the sky is blue        [noun, color]
       the apple is red.      [noun, color]
       Let's take a walk      [verb]

I have tried:

data.loc[data.body.str.contains("red|blue"), 'label'] = data.label.str.append('color')

akuiper · Accepted Answer · 2021-09-23 18:26:21Z

2

One option is to use apply on the Series and then directly append to list:

data.loc[data.body.str.contains('red|blue'), 'label'].apply(lambda lst: lst.append('color'))

data
                body          label
0    the sky is blue  [noun, color]
1  the apple is red.  [noun, color]
2  Let's take a walk         [verb]

answered Sep 23, 2021 at 18:26

akuiper

216k33 gold badges362 silver badges379 bronze badges

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

2 Comments

mjoy Over a year ago

this works. thanks!

akuiper Over a year ago

@mpatil Cool. Glad to hear!

Collectives™ on Stack Overflow

How to append value to list in a row based on another column python?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related