파이썬/Basic

파이썬 답게 코딩해보자 if condition

코샵 2022. 11. 13. 00:41
반응형

pytorch를 공부하다가 아래 코드를 무작정 가져다 쓰기만 했다. 

그러다 문득 변수 초기화할때 if문으로 변수를 지정하는것이 눈에 띄었다.

device = 'cuda' if torch.cuda.is_available() else 'cpu'

대충 이해는 되었지만 정확히 알고 있는게 아니라 실제 코딩할때 자연스럽게 쓰이지 않더라 

 

num = 0
state = '양수' if num > 0 else '음수' 
print(state)

변수 초기화 할때 두가지 경우의 수가 있다면 위와 같이 사용하면 좋을거 같다. 

다만 elif 는 안되는거 같다..... 


여기서 잠깐 if문 미세 팁을 드리자면

a = 1 
if a > 0 : print('양수 입니다')
elif a == 0 : print('0입니다')
else : print('음수 입니다')

조건식의 실행 문장이 한줄일 경우 위와 같이 줄내림하지않고 사용할 수 있다. 

나는 이게 보기 편하고 코드가 이뻐보여서 이렇게 사용한다. 줄내림을 해도되고 안해도 되는 취향차이