Reset index on multiple dataframes

Question

What I want to do is write a line of code that resets in the indices on some data frames. I thought I could write something like this

X_train, X_test_, y_train, y_test = train_test_split(X,y)
map(lambda x: x.reset_index(drop=True,inplace=True),
                 [X_train, X_test_, y_train, y_test])

But it doesn't have the desired result. Suggestions?

PMende · Accepted Answer · 2019-10-15 19:15:03Z

2

map is a lazy operation. It doesn't run until you iterate over the map. You can achieve your desired result by simply running a list comprehension:

X_train, X_test_, y_train, y_test = [
    df.reset_index(drop=True)
    for df in train_test_split(X,y)
]

answered Oct 15, 2019 at 19:15

PMende

5,4903 gold badges21 silver badges26 bronze badges

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

Collectives™ on Stack Overflow

Reset index on multiple dataframes

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