Pandas conditional concatenate of a dataframe column

Question

This question is an extension of Pandas conditional creation of a series/dataframe column

Given the below dataframe:

df = pd.DataFrame(np.random.randint(0,20,size=(10, 2)), columns=list('AB'))
df["note"]="old note"

I need to update A and note columns when A>B. This is what I tried:

df.loc[df.A>df.B, "note"]="A was: "+str(df.A)
df.loc[df.A>df.B, "A"]=df.B

and

df["note"]=np.where(df.A>df.B, "A was: "+str(df["A"]), df["note"])
df.loc[df.A>5, "A"]=df.B

the note column does not come out clean, it concatenates the entire A column in a single cell, while I would like to keep only the value of the same row. could you please help?

jezrael · Accepted Answer · 2019-12-03 11:31:09Z

2

Here is problem with casting column to strings, use Series.astype:

df.loc[df.A>df.B, "note"]="A was: "+df.A.astype(str)

edited Dec 3, 2019 at 11:31

answered Dec 3, 2019 at 11:28

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

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

Collectives™ on Stack Overflow

Pandas conditional concatenate of a dataframe column

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related