用条件替换单词

Question

I would need to check if a row contains a number (n), a blank space and a word in the list (house, houses, casa, case), in order to replace it with

• 10.00 if the number is equal to 1 ;
• 10.00*n where n is a number not equal to 1 .

Example:

H_C
4 case        
9 apart          
1 house    

Expected output

H_C              New_H_C
4 case          40.00  
9 apart          9 apart
1 house          10.00

I have tried using a mix of re.search looking at the list (house, houses,case,casa), and strip for adding .00 , but I am confused on how to use it to determine the conditions 10.00 and 10.00*n .

I hope you can help me.

Answer 1

You can do this regex:

words = ['house', 'houses', 'casa', 'case']

# extract the relevant numbers
s = df.H_C.str.extract(f'(\d+) ({"|".join(words)})')[0].astype(float)
# multiply with 20

# you can search for `format` string function
df['New_H_C'] = np.where(s.notna(),s.apply(lambda x:'{:.02f}'.format(x)), 
                         df['H_C'])

Output:

       H_C  New_H_C
0   4 case     4.00
1  9 apart  9 apart
2  1 house     1.00