different cells for if elif else in colab

Question

I am wondering if there is any way to use if-statement in one cell, then use elif in another cells, then use else in the final cell, in go