How To Create A 2d Array In Python Using And In If Statement