How to remove certain string from column in pandas dataframe

Question

I want to remove a certain keywords or string in a column from pandas dataframe.

The dataframe df looks like this:

YEAR    WEEK
2019    WK-01
2019    WK-02
2019    WK-03
2019    WK-14
2019    WK-25
2020    WK-06
2020    WK-07

I would like to remove WK-and 0 from the WEEK column so that my output will looks like this:

YEAR    WEEK
2019    1
2019    2
2019    3
2019    14
2019    25
2020    6
2020    7

Quang Hoang · Accepted Answer · 2020-06-25 13:46:44Z

3

You can try:

df['WEEK'] = df['WEEK'].str.extract('(\d*)$').astype(int)

Output:

   YEAR  WEEK
0  2019     1
1  2019     2
2  2019     3
3  2019    14
4  2019    25
5  2020     6
6  2020     7

answered Jun 25, 2020 at 13:46

Quang Hoang

151k11 gold badges64 silver badges86 bronze badges

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

Comments

timgeb · Accepted Answer · 2020-06-25 13:47:38Z

3

Shave off the first three characters and convert to int.

df['WEEK'] = df['WEEK'].str[3:].astype(int)

answered Jun 25, 2020 at 13:47

timgeb

79.2k20 gold badges129 silver badges150 bronze badges

3 Comments

Nur Atiqah Over a year ago

what's the different between using the str.extract and only str?

timgeb Over a year ago

@NurAtiqah str.extract lets you extract matches from regular expressions. str[...] allows vectorized access to specific elements or slices of elements.

Nur Atiqah Over a year ago

noted...so..if I were to combine this output to the original dataframe that has multiple column...I can use join() right?

Collectives™ on Stack Overflow

How to remove certain string from column in pandas dataframe

2 Answers 2

Comments

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related