What does assign(home=1) mean in Pandas Python?

Question

I am trying to understand what the following snippet of the code does, in particular the assign(home=1).rename part:

goal_model_data = pd.concat([epl[['HomeTeam','AwayTeam','HomeGoals']].assign(home=1).rename(
            columns={'HomeTeam':'team', 'AwayTeam':'opponent','HomeGoals':'goals'}),
           epl[['AwayTeam','HomeTeam','AwayGoals']].assign(home=0).rename(
            columns={'AwayTeam':'team', 'HomeTeam':'opponent','AwayGoals':'goals'})])

This is taken from text

That deals with regression modelling in pandas. I don't exclude the possibilities of mistakes in the code snippets they provide. But when I run the code, it seems to function properly. But I don't actually understand the meaning of the above given code and how it functions.

Please help.

Looks to be creating a binary column, which will have a value of 0 for the first df and 1 for the second — Chris
– Chris, Commented May 22, 2024 at 14:05
Ok, but if it's new column named "home", why doesn't the name inside assign come enclosed in quotation marks? I would expect it to be assign("home"=1), not assign(home=1) — Astral
– Astral, Commented May 22, 2024 at 14:32
DataFrame.assign(**kwargs), it's like doing dict(home=1) (equivalent to {"home": 1}). — Timeless
– Timeless, Commented May 22, 2024 at 15:35
So, df.assign(home=1) is equivalent to df.assign(**{"home": 1}). — Timeless
– Timeless, Commented May 22, 2024 at 15:37

amance · Accepted Answer · 2024-05-22 14:25:26Z

0

That uses pd.df.assign() to add a new column with specific values. In the case of assign(home=1), it adds a column called 'home' where every row will have the value of 1.

answered May 22, 2024 at 14:25

amance

1,8528 silver badges18 bronze badges

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

2 Comments

Astral Over a year ago

Why isn't "home" inside assign() in quotation marks?

amance Over a year ago

I'm not sure why they wrote the code that way. Please see the examples at pandas.pydata.org/docs/reference/api/…

Collectives™ on Stack Overflow

What does assign(home=1) mean in Pandas Python?

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related