Python：如何有条件地从python中的数据框中的列中删除字母？

Question

data = {'period':['chy1md','chy2md','chy6md',chy6L6L1y,'chy6L6L5y','chy6L6L10y']}
df = pd.DataFrame(data)

Expected output:

new_df = {'period':['1md','2md','6md',1y,'5y','10y']}

May I get help to get the above expected output. I want conditional strip from the column based on length of the column values. I dont want to hard code the letters to be removed. It is so because I have many files and there are different names in the columns. Conditional stripping help me to format other files.

Answer 1

Do:

result = df['period'].str.replace('(chy|6L6L)', '')
print(result)

Output

0    1md
1    2md
2    6md
3     1y
4     5y
5    10y
Name: period, dtype: object

Answer 2

IIUC, this woudl do:

df.period.str.extract('(\d+\D+)$')

Output:

     0
0  1md
1  2md
2  6md
3   1y
4   5y
5  10y