Indexing a DataFrame

Question

I am new to Python and I am trying to use indexing to obtain the last name along each row.

import numpy as np import pandas as pd def manager(vec): for i,val in enumerate(vec): if val == np.NaN: break return vec[i - 1] df = pd.DataFrame({'ID':[23,15,20], 'L1_name': ['Andrew','Thomas','Thomas'], 'L1_ID': [234,994,994],'L2_Name':['Andrew','Alice','Thomas'],'L2_ID':[234,237,994], 'L3_Name':['Jerico','Sarah',np.nan],'L3_ID':[453,237,np.nan]}) df['new'] = df.apply(manager, axis=1) df

What am i doing wrongly, anyone?

your code ist probably not correctly formated. Your function manager returns in the first round of the foor loop, either with vec[- 1] or None. — DiCaprio
– DiCaprio, Commented Feb 20, 2019 at 12:15

jezrael · Accepted Answer · 2019-02-20 12:10:36Z

Use filter with forward filling per rows of missing values with seelct last column by positions:

s = df.filter(like='Name').ffill(axis=1).iloc[:,-1] print (s) 0 Jerico 1 Sarah 2 Thomas Name: L3_Name, dtype: object

heyu91 · Accepted Answer · 2019-02-20 12:02:19Z

0

I think this works

def manager(vec): return vec.dropna().iloc[-2]

answered Feb 20, 2019 at 12:02

heyu91

1,28212 silver badges19 bronze badges

Collectives™ on Stack Overflow

Indexing a DataFrame

2 Answers 2

Comments

Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Related