Replace string with one word based on condition in python

Question

In a pandas dataframe, I have a string column with multiple values and I want to replace it one with based on a match for different rows.

Eg - Based on image, I want to specify "Extreme progressive,Progressive rock,Progressive" as Progressive, "Heavy,Bay area thrash" as Thrash, "Progressive death,Death,Progressive thrash" as Death and so on. How should I proceed with executing the same?

B Man · Accepted Answer · 2019-10-14 19:26:13Z

1

Using a dataframe:

dfa:

        ID No   Time    Variable    Val
0       123     0.1     A           1
1       123     0.1     B           2
2       123     0.1     C           3
3       127     0.8     A           4
4       127     0.8     B           5
5       127     0.8     C           6

you can create a dict mapper:

dictMapper = {'A' : 'aye', 'B': 'bee'}
dfa['Variable'] = dfa['Variable'].map(lambda x: dictMapper.get(x,x))
dfa:
    ID No   Time    Variable    Val
0   123     0.1     aye         1
1   123     0.1     bee         2
2   123     0.1     C           3
3   127     0.8     aye         4
4   127     0.8     bee         5
5   127     0.8     C           6

This ofcourse relies on you knowing all before: after combinations before updating your dataframe as you'll need an exact match to the dict keys.

answered Oct 14, 2019 at 19:26

B Man

994 bronze badges

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

2 Comments

sklal Over a year ago

This works, though all the necessary combinations are required in this case. How to proceed if one decides on replacing the string with a word based on a match in the string?

B Man Over a year ago

you'll still need some form of maping dict I assume. You could split the string on commas and check each word against the dict.keys

Collectives™ on Stack Overflow

Replace string with one word based on condition in 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