Multiple special character transformations on dataframe using Pandas

Question

I wish to keep everything before the hyphen in one column, and keep everything before the colon in another column using Pandas.

Data

ID            Type               Stat

AA - type2    AAB:AB33:77:000    Y
CC - type3    CCC:AB33:77:000    N

ID    Type

AA    AAB
CC    CCC

separator = '-'
result_1 = my_str.split(separator, 1)[0]

Any suggestion is appreciated

`df[['ID', 'Type']] = df[['ID', 'Type']].apply(lambda col: col.str.extract(r'(\w+)', expand=False), axis=0)` — Trenton McKinney, Dec 09 '22 at 00:09

score 1 · Answer 1 · answered Dec 08 '22 at 23:30

1

I would say

func1 = lambda _: _['ID'].split('- ')[0]
func2 = lambda _: _['Type'].split(':')[0]

data\
  .assign(ID=func1)\
  .assign(Type=func2)

References https://pandas.pydata.org/docs/reference/api/pandas.DataFrame.assign.html

answered Dec 08 '22 at 23:30

rafidini

score 1 · Accepted Answer · answered Dec 08 '22 at 23:38

1

We can try using str.extract here:

df["ID"] = df["ID"].str.extract(r'(\w+)')
df["Type"] = df["Type"].str.extract(r'(\w+)')

answered Dec 08 '22 at 23:38

Tim Biegeleisen