[Python] 코딩테스트를 위한 파이썬 문법 정리 (2) 조건문, 반복문, 함수

이것이 취업을 위한 코딩테스트다 with 파이썬

이것이 취업을 위한 코딩 테스트다 with 파이썬 - YES24

2. 조건문

- if ~ elif ~ else 문 이용

if 조건문1:
	조건문 1이 True일 때 실행되는 코드
elif 조건문2:
	조건문 1에 해당하지 않고, 조건문 2가 True일 때 실행되는 코드
else:
	위의 모든 조건문이 모두 True 값이 아닐 때 실행되는 코드

- 예제1)

score = 85

if score >= 90:
	print("학점: A")
elif score >= 80:
	print("학점: B")
elif score >= 70:
	print("학점: C")
else:
	print("학점: F")
    
# 출력 -> 학점: B

비교 연산자

X == Y

X != Y

X > Y

X < Y

X >= Y

X <= Y

논리 연산자

X and Y : X와 Y가 모두 참일 때 참

X or Y : X와 Y 중에 하나만 참이어도 참이다 

not X : X가 거짓일 때 참이다

기타 연산자

X in 리스트 : 리스트 안에 X가 들어가 있을 때 참이다

X not in 문자열 : 문자열 안에 X가 들어가 있지 않을 때 참이다

 

 

- 조건부 표현식을 이용하면 if ~ else문을 한 줄에 작성해 사용할 수 있다.

score = 85
result = "Success" if score >= 80 else "Fail"

print(result)

# 출력 -> Success

- 예제2) 리스트에서 간결하게 사용하기

a = [1, 2, 3, 4, 5, 5, 5]
remove_set = {3, 5}

result = []
for i in a:
	if i not in remove_set:
    	result.append(i)
        
print(result)

# 출력 -> [1, 2, 4]


######간단하게 작성######

result = [i for i in a if i not in remove_set]
print(result)

# 출력 -> [1, 2, 4]

---

3. 반복문

 while문

- 조건문이 참일 때에 한해서, 반복적으로 코드가 수행된다.

i = 1
result = 0

# i가 9보다 작거나 같을 때 아래 코드를 반복적으로 실행
while i <= 9:
	result += i
    i += 1

print(result)

# 출력 -> 45

for문

- in 뒤에 오는 데이터에 포함되어 있는 모든 원소를 첫 번째 인덱스부터 차례대로 하나씩 방문한다. in 뒤에 오는 데이터로는 리스트, 튜플, 문자열 등이 사용될 수 있다.

for 변수 in 리스트:
	실행할 소스코드

- 예제(위의 while문과 동일)

result = 0

# i는 1부터 9까지의 모든 값을 순회
for i in range(1, 10):
	result += i

print(result)

# 출력 -> 45

- range()의 값으로 하나의 값만 넣으면, 자동으로 시작 값은 0이 된다.

- continue 사용시 반복문의 처음으로 돌아간다.

- 중첩 사용 가능, 플로이드 워셜 알고리즘, 다이나믹 프로그래밍 등에서 사용된다.

---

4. 함수

def 함수명(매개변수):
	실행할 소스코드
    return 반환 값

- 예시

def add(a, b):
	return a + b
    
print(add(3, 7))

# 출력 -> 10

-  파라미터의 변수를 직접 지정해서 값을 넣어 주는 것이 가능하다. 매개변수의 순서가 달라도 상관없음

def add(a, b):
	print('함수의 결과:', a + b)

add(b = 3, a = 7)

# 출력 -> 함수의 결과: 10

- 함수 안에서 함수 밖의 변수 데이터를 변경해야하는 경우에는 global 키워드로 변수를 지정해서 사용한다.

a = 0

def func():
	global a
    a += 1
    
for i in range(10):
	func()
    
print(a)

# 출력 -> 10

- 간단한 함수는 람다 표현식을 사용하여 작성가능

def add(a, b):
	return a + b
    
# 일반적인 add()메서드 사용
print(add(3, 7))

# 람다 표현식으로 구현한 add() 메서드
print((lambda a, b: a + b)(3, 7))

# 출력 -> 
# 10 
# 10