[파이썬] 내가 보려고 만든 파이썬 팁

 

알고리즘을 풀기 위한 꿀팁 모음!!

 

 

1. 일렬로 배열 출력

print("".join(배열))

• 배열을 문자열 형식으로 일렬로 출력

 

 

2. 문자열 확인

if '3' in str(i) + str(j) + str(k):
    count += 1

• 위 코드는 i, j, k를 문자열화 시켜서 하나로 합친 뒤, 그 안에 '3'이 하나라도 있으면 count에 1을 더하는 것이다.

 

 

3. int형 입력 여러개 받기

data = list(map(int, input().split()))

• 리스트 형식으로 받기

n, m = map(int, input().split())

• 다른 변수에 각각 받기

n = int(input())
array = [0] * 10_000
for i in range(n):
    array[i] = int(input())

• 백준에서 여러줄 입력받을 때 사용

 

 

4. 배열 정렬하기

arr.sort()

• sort() 함수 사용, arr의 내용이 자동으로 정렬됨(오름차순)

arr.sort(reverse=True)

• 거꾸로 정렬(내림차순)

 

 

5. 이진 탐색

from bisect import bisect_left, bisect_right

bisect_left(a, x)

•  정렬된 순서를 유지하면서 배열 a에 x를 삽입할 가장 왼쪽의 인덱스를 반환
•  C++ 에서 lower_bound()와 동일

bisect_right(a, x)

• 정렬된 순서를 유지하면서 배열 a에 x를 삽입할 가장 오른쪽의 인덱스를 반환
•  C++ 에서 upper_bound()와 동일

 

 

6. 2차원 배열 만들기

arr = [[0] * M for _ in range(N)]

• 행이 M개, 열이 N개인 2차원 배열이 만들어짐(내용은 전부 0)

arr = [[] for _ in range(n)]

• 내용이 없는 n개 행을 가진 2차원 배열

arr = []
for N in range(N):
    arr.append(list(map(int, input().split())))

• 띄어쓰기 되어 있는 문자열을 2차원 배열로 입력 받을 때

arr = []
for N in range(N):
    arr.append(list(map(int, input())))

• 띄어쓰기가 안 되어있는 입력을 받을 때 2차원 배열을 만드는 법
• input() 뒤에 split()만 빼줘도 된다.

n, m = map(int, input().split())         # 행과 열
array = list(map(int, input().split()))  # 일렬로 입력
dp = []
index = 0
for i in range(n):
    dp.append(array[index:index+m])
    index += m

• 2차원 배열을 일렬로 입력받을 때 2차원으로 만드는 방법
• 슬라이싱을 이용한다.

 

 

 

7. 알파벳인지 숫자인지 확인하기

A = "1234"
B = "A23BD"
C = "ABD"
print(A.isalpha())       # False
print(A.isdigit())       # True
print(B.isalpha())       # False
print(B.isdigit())       # False
print(C.isalpha())       # True
print(C.isdigit())       # False

• isalpha() 와 isdigit() 사용

 

 

8. 리스트.count()

num = list형자료.count(0)

• 0의 개수를 리턴해준다.

 

 

9. 정규 자료형 확인

문자열.isalnum()

• 문자, 숫자로 구성되어 있으면 true, 아니면 false