Getting first values with index in Series

Question

Given

import pandas as pd
df = pd.DataFrame({'lowercase':['a','b','c','d']},
                       index=[0,1,3,4])

# print(df)

print(df['lowercase'].iat[0])

I can get 'a', but I want to get the first index (which is 0) and 'a' return as two values, how should I do?

df.index[0] ? what do you mean by and 'a' return as two values, ? — anky
– anky, Commented Jun 14, 2020 at 14:16
Thanks @ScottBoston @anky a=df['lowercase'].index[0] b=df['lowercase'].iat[0] Can this be simplify in one line? — william007
– william007, Commented Jun 14, 2020 at 14:18
a, b = df['lowercase'].index[0], df['lowercase'].iat[0]? It's one line :) — Henry Yik
– Henry Yik, Commented Jun 14, 2020 at 14:22

Ch3steR · Accepted Answer · 2020-06-14 15:11:02Z

1

You can use df.itertuples which returns a map object.

def get_item(idx):
    it = df.itertuples()
    while idx:
        next(it)
        idx = idx-1
    return next(it)

a,b = get_item(0)
# a = 0 , b = 'a'
c,d = get _item(1)
# c = 1, d = 'b'

You can even get a slice using islice from itertools.

def get_item(start,end=None):
    it = df.itertuples()
    if end is None:
        while idx:
            next(it)
            idx = idx-1
        return next(it)
    return list(map(tuple,islice(it,start,end)))

get_item(1, 4)
# [(1, 'b'), (3, 'c'), (4, 'd')]

If you only want first value

a,b = next(df.itertuples())
# a = 0, b = 'a'

edited Jun 14, 2020 at 15:11

answered Jun 14, 2020 at 14:37

Ch3steR

20.8k4 gold badges34 silver badges66 bronze badges

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

Collectives™ on Stack Overflow

Getting first values with index in Series

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related