Pandas 从 dataframe 中的列中提取部分字符串并将其存储在新列中

Question

I have the below code to create a dataframe

import pandas as pd
df = {'Connection':['Home 10Mbps','Broadband 5 Mbps','128 Kbps Internet','Discounted 512Kbps 2 years contract']}
df = pd.DataFrame (df)
df

I need a way to extract only the bandwidth from column "Connection" and store the result in new column named for example "Bandwidth" to look something like this:

Bandwidth

10 Mbps

5 Mbps

128 Kbps

512 Kbps

Answer 1

Make sure to fill the list with all possible formats

lst = ['10Mbps', '10 Mbps', '5 Mbps', '128 Kbps', '512Kbps', '512 Kbps']
for i in lst:
   df.loc[df['Connection'].str.contains(i), 'bandwidth'] = i

# output
Connection                 bandwidth
Home 10Mbps                 10Mbps
Broadband 5 Mbps            5 Mbps
128 Kbps Internet           128 Kbps
Discounted 512Kbps 2 years contract 512Kbps