파이썬 코드 개선하기

1️⃣ one line으로 사용하기

💩 Before

for rate in failureRate:
	answer.append(rate[0])

👍 After

answer = [i[0] for i in failureRate]

+) one line if else statement

print("A") if a > b else print("B")

2️⃣ count 함수

EX) 리스트 list = [1,2,2,3,4,5,1,4] 안에서 1이 몇갠지, 2가 몇갠지 세야 하는 경우 (번호가 N=5까지 있을 때)

💩 Before

answer = [[0] for i in range(N)]
for i in list:
	answer[i-1] += 1

👍 After

answer = []
for i in range(1,N+1):
	count = list.count(i)
    answer.append(count)

3️⃣ 1차원 배열 2차원 배열에 옮기기

💩 Before

dp = []
for _ in range(n):
    dp.append([0 for _ in range(m)])
for i in range(n):
    for j in range(m):
        dp[i][j] = inputs[m*i+j]

👍 After

dp = []
index = 0
for i in range(n):
    dp.append(inputs[index:index + m])
    index += m