python lamba - if statement with conditions in multiple columns

ben121 :

I have a dataframe:

df = pd.DataFrame({'col1': [1,2,3,4,5], 'col2':[1,2,3,4,5], 'col3':['a','b','c','d','e']})

I want to create a new column that says 'yes' if col1 and col2 are equal to 1 else it would say no.

I have tried using a lambda function but no success:

df['win'] = df[['col1', 'col2']].apply(lambda x: 'Yes' if (x['col1'] == 1) & (x['col2'] == 1) else 'No')

Is there a better way of doing this? or an improvement on what I have done so it works.

HS-nebula :

Yes, you can use np.where.

import pandas as pd
import numpy as np

df = pd.DataFrame({'col1': [1,2,3,4,5], 'col2':[1,2,3,4,5], 'col3':['a','b','c','d','e']})

condition = ((df['col1'] == 1) & (df['col2'] == 1))
# Return 'Yes' if the condition is True, and 'No' if False
df['win'] = np.where(condition, 'Yes', 'No')

Guess you like

Origin http://43.154.161.224:23101/article/api/json?id=360458&siteId=1

python lamba - if statement with conditions in multiple columns

How to group by multiple columns and create a new column based on conditions in Python?

Multiple conditions in an SQL Delete statement

Java If Statement having multiple conditions with && operator not working

Moving rows in multiple columns into their own columns Python

Selecting dataframe rows based on multiple columns, where new functions should be created to handle conditions in some columns

Sql statement, cut a piece of data in the database into multiple columns through delimiters

Detailed explanation of the use of CASE WHEN function statement under multiple conditions

Three methods of parallel judging python if multiple conditions

Robot Framework [string] is determined, IF statement is executed multiple conditions, multiple execution statements

Python one line of content becomes multiple columns (one line becomes multiple columns)

Python virtuoso operation: seven kinds of conditions written statement

Ways to iterate over multiple iterables in a Python for statement

Where in the mybatis statement must be followed by necessary conditions and multiple selection condition processing methods

How to groupby multiple columns and create a new column in Python based on thresholds

"argument of type 'int' is not iterable" appears when matching multiple conditions in Python

Binding lock multiple conditions

Judgment of multiple conditions in the applet

OR to trigger multiple conditions

Learning python (9)-closure function and lamba

python cycling conditions - conditions

Conditions of sql statement, paging, sorting

QListView implements flow layout and multiple columns (flow layout and multiple columns)

How to select a group by multiple columns