How to apply formula on a column in Dataframe based on value from other column
Question
I have a DataFrame and I want to apply formula on Column A based on Column B.
If Column B has some values then apply formula on column A ( 100 - column value ). Below is the data I have.
DataFrame Input
A B
0 35 TYLER
1 20 MACK
2 30 MACK
3 40 MITCH
If column B has 'MACK' and 'MITCH' then apply formula, else not. How can I do that?
DataFrame Output
A B
0 35 TYLER
1 80 MACK
2 70 MACK
3 60 MITCH
1 answers
solution1
0 ACCPTED 2022-03-02 21:34:19
You can limit your formula to just the matching rows with boolean indexing:
apply_rows = df["B"].isin(["MACK", "MITCH"])
df.loc[apply_rows, "A"] = 100 - df.loc[apply_rows, "A"]
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.
Related Question
apply function on subset of dataframe rows in column based on value in other column
Pandas dataframe apply function to column strings based on other column value
Apply function to dataframe column based on combinations of values from other columns
Adding new Column(s) to a dataframe based on value from other column
How to fill column based on value of other column in dataframe?
Add column to dataframe based on value in other column
Modifying value in dataframe column based on other column
Apply function to dataframe based on column with other dataframe based on index
Apply function to dataframe column element based on value in other column for same row?
Getting a value from DataFrame based on other column value (PySpark)
Related Tags