pandas list comprehension code example

Example 1: pandas list comprehension

# Subtract 1 from row, for each row in df.year
df['previous_year'] = [row-1 for row in df['year']]

Example 2: comprehension list iloc pandas

df['Flag2'] = ["Increase" if x > y  else "Decrease" for (x,y) in zip(df['nextsalary'],df['prevsalary'])]

Example 3: comprehension list iloc pandas

df['Flag'] = ["High Bracket" if x > 70 else "Low Bracket" for x in df['prevsalary']]

Example 4: comprehension list iloc pandas

mat = [[1,2], [3,4], [5,6]]
[x for row in mat for x in row if x%2==1]