Python - Pandas - Replace a string from a column based on the value from other column

Question

I've a dataframe with the following data:

d = {'col0': ['Table 1', 'Table 2'], 'col1': ['Tablename: Table 1', 'Tablename: Table 2'], 'col2': ['Table A', 'Table B']}

What I am trying is to replace on col2 the values that exists on col0 with the values from col2.

Basically my output it will be this:

I was trying to make a simple replace such as:

df['col3'] = df['col1'].replace(df.col0, df['col2'], regex = True)
print(df)

But it gives me "TypeError: replace() takes no keyword arguments"

Then I try this:

df['col3'] = df['col1'].replace(df.col0, str(df['col2']), regex = True)
print(df)

But it gives me a list on col3...

How can I do that?

Thanks

Yuna A. · Accepted Answer · 2020-02-13 17:26:39Z

1

You have to pass a list or something list-like to the first and second parameters of replace. .values turns df['col0'] and df['col2'] into numpy arrays which will work.

df['col3'] = df['col1'].replace(df['col0'].values, df['col2'].values, regex = True)

Output:

      col0                col1     col2                col3
0  Table 1  Tablename: Table 1  Table A  Tablename: Table A
1  Table 2  Tablename: Table 2  Table B  Tablename: Table B

You can see the full documentation of pd.DataFrame.replace here.

edited Feb 13, 2020 at 17:26

answered Feb 13, 2020 at 17:16

Yuna A.

1496 bronze badges

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

Collectives™ on Stack Overflow

Python - Pandas - Replace a string from a column based on the value from other column

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related