create a new column in pandas dataframe using if condition

# Method 1: df.loc[df['column name'] condition, 'new column name'] = 'value if condition is met' #or df.loc[df['set_of_numbers'] <= 4, 'equal_or_lower_than_4?'] = 'True' # Method 2: df['new column name'] = df['column name'].apply(lambda x: 'value if condition is met' if x condition else 'value if condition is not met') #or df['name_match'] = df['First_name'].apply(lambda x: 'Match' if x == 'Bill' else 'Mismatch') # or df.loc[(df['First_name'] == 'Bill') | (df['First_name'] == 'Emma'), 'name_match'] = 'Match' df.loc[(df['First_name'] != 'Bill') & (df['First_name'] != 'Emma'), 'name_match'] = 'Mismatch'

# For creating new column with multiple conditions conditions = [ (df['Base Column 1'] == 'A') & (df['Base Column 2'] == 'B'), (df['Base Column 3'] == 'C')] choices = ['Conditional Value 1', 'Conditional Value 2'] df['New Column'] = np.select(conditions, choices, default='Conditional Value 1')

# Create a new column called based on the value of another column # np.where assigns True if gapminder.lifeExp>=50 gapminder['lifeExp_ind'] = np.where(gapminder.lifeExp >= 50, True, False) gapminder.head(n=3)

python conditionally create new column in pandas dataframe

# If you only have one condition use numpy.where()
# Example usage with np.where:
df = pd.DataFrame({'Type':list('ABBC'), 'Set':list('ZZXY')}) # Define df
print(df)
  Type Set
0    A   Z
1    B   Z
2    B   X
3    C   Y

# Add new column based on single condition:
df['color'] = np.where(df['Set']=='Z', 'green', 'red')
print(df)
  Type Set  color
0    A   Z  green
1    B   Z  green
2    B   X    red
3    C   Y    red


# If you have multiple conditions use numpy.select()
# Example usage with np.select:
df = pd.DataFrame({'Type':list('ABBC'), 'Set':list('ZZXY')}) # Define df
print(df)
  Type Set
0    A   Z
1    B   Z
2    B   X
3    C   Y

# Set the conditions for determining values in new column:
conditions = [
    (df['Set'] == 'Z') & (df['Type'] == 'A'),
    (df['Set'] == 'Z') & (df['Type'] == 'B'),
    (df['Type'] == 'B')]

# Set the new column values in order of the conditions they should
#	correspond to:
choices = ['yellow', 'blue', 'purple']

# Add new column based on conditions and choices:
df['color'] = np.select(conditions, choices, default='black')

print(df)
# Returns:
  Set Type   color
0   Z    A  yellow
1   Z    B    blue
2   X    B  purple
3   Y    C   black

Posted by: Guest on November-12-2020

Source

Code answers related to "Python"

Browse Popular Code Answers by Language

Answers for "create a new column in pandas dataframe using if condition"

Code answers related to "create a new column in pandas dataframe using if condition"

Code answers related to "Python"

Python Answers by Framework

Browse Popular Code Answers by Language

Popular Programming Languages

Advertisements

Company

Compilers

Help

Connect with us