Create new dataframe column based on a specific character of a string in another column, pandas

Question

I have a dataframe with a column that contain strings. I want to know if it is possible to create a new column based on the dataframe. This is an example of the column:

   col1
016e3d588c
071b4x718g
011e3d598c
041e0i608g

I want to create a new column based on the last character of the string. This is what I tried:

for i in DF['col1']:
    if i[-1] == 'g':
        DF['col2'] = 1
    else:
        DF['col2'] = 0

I want the new column like this:

but my code have the following output:

Is possible to do it?

Thanks in advance

AMH · Accepted Answer · 2020-06-10 12:26:29Z

1

You can try this using numpy.

import numpy as np

DF["col2"] = np.where(DF["col1"].str[-1]=="g",1,0)

answered Jun 10, 2020 at 12:26

AMH

5282 silver badges13 bronze badges

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

Comments

Rakesh · Accepted Answer · 2020-06-10 12:24:56Z

1

Using str.endswith()

Ex:

df = pd.DataFrame({"Col1": ['016e3d588c', '071b4x718g', '011e3d598c', '041e0i608g']})
df["Col2"] = df["Col1"].str.endswith("g").astype(int)
print(df)

Output:

         Col1  Col2
0  016e3d588c     0
1  071b4x718g     1
2  011e3d598c     0
3  041e0i608g     1

answered Jun 10, 2020 at 12:24

Rakesh

82.9k17 gold badges85 silver badges122 bronze badges

Collectives™ on Stack Overflow

Create new dataframe column based on a specific character of a string in another column, pandas

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related