I want to change the value of a variable

How to change the value of a variable using if, elif, else

  a = 0
  b = 321
  if b == 321:
    a == 2

print(a)

I want it to be exactly 2