python how to add IF statement in a map function using lambda

Question

I have this following df:

d = {'role': ['admin/tech', 'admin', 'programming/asst/design']}
df = pd.DataFrame(data=d)

My goal is:

In other words, extract everything before the first '/' IF row contains '/'

I came up with:

and now I need to add a IF statement for this function

I tried something like:

But it doesn´t work.

Thanks in advance, I really appreciate suggestions!!

Python's conditional expressions need an else statement, hence the error. Otherwise what should happen if x.lower is not True? docs.python.org/2.5/whatsnew/pep-308.html — Random Davis
– Random Davis, Commented Jul 13, 2021 at 16:05
I strongly believe it's some IDE error. the code on terminal is different from code on text editor. delete and rewrite the line might work. — fahmiduldul
– fahmiduldul, Commented Jul 13, 2021 at 16:06

not_speshal · Accepted Answer · 2021-07-13 16:05:45Z

2

You don't need a lambda function for what you want to achieve. Try this:

>>> df["role"].str.split("/").str[0]
0          admin
1          admin
2    programming
Name: role, dtype: object

If for whatever reason you still want to use lambda, do this:

>>> df["role"].apply(lambda x: x.split("/")[0])
0          admin
1          admin
2    programming
Name: role, dtype: object

answered Jul 13, 2021 at 16:05

not_speshal

23.2k2 gold badges18 silver badges33 bronze badges

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

1 Answer 1