코딩테스트를 위한 파이썬 메서드

1. 출력 - print()

터미널에 데이터를 출력한다.

print(3) # 3
print(type(3)) # int
print(3, 6) # 3 6
print(3, end=' ')
print(5) # 3 5

 

2. 입력 - input()

사용자의 입력을 받는다.파이썬 인터프리터는 input()을 만나면 멈추고 사용자의 입력을 기다린다. input을 사용하면 사용자 인풋이 스트링으로 변환된다.

num = input()
num2 = input('입력해주세요:')

여러 개의 입력

공백 기준으로 여러 입력이 주어지면, int(), map(함수, iterable)을 활용해 변환한다.

# input이 1 2 3 4 5 6일 때, 
# 결과적으로 [1, 2, 3, 4, 5, 6] 이라는 리스트 안의 int자료형으로 변환하고 싶다면?
nums = list(map(int, input().split()))
print(nums) # [1, 2, 3, 4, 5, 6]

• input : '1 2 3 4 5 6'
• input.split() : input을 쪼개었으나 원소들이 모두 스트링임, ['1', '2', '3', '4', '5', '6']
• map(int, input().split()) : 원소 int로 map
• list(map(int,input().split()) : map을 list로 변환

2차원 리스트 받기

# 1 2 3 => 1 2 3 을 입력후 엔터
# 4 5 6 => 그다음 4 5 6 입력후 엔터
# 7 8 9 => 그다음 7 8 9 입력후 엔터

nums_matrix = [list(map(int, input().split())) for _ in range(3)]
print(nums_matrix, type(nums_matrix))
# [[1, 2, 3], [4, 5, 6], [7, 8, 9]] <class 'list'>

 

3. 리스트

• 리스트는 기본적으로 대괄호로 표현하며, 인덱스 값을 활용해 O(1)시간 안에 접근할 수 있다.
• 없는 인덱스를 지정하면, IndexError가 발생한다.
• 리스트 안에는 다양한 자료형이 들어갈 수 있고, 길이는 가변적이다.

.append()

리스트 맨 뒤에 요소를 1개 추가한다.

.pop(index)

리스트에서 인덱스 위치에 해당하는 원소를 추출한 후 반환한다.

.sort()

작은 수부터 차례로 정렬한다. key 옵션을 통해 기준을 선택할 수 있다.

nums = [3, 5, 1, 4, 2]
nums.sort()
print(nums) # [1, 2, 3, 4, 5] => 원본 자체를 정렬한다.

# 만약 역순으로 정렬하고 싶은 경우? key 옵션과 익명함수 lambda를 활용한다.
# nums.sort(key=lambda x: -x)

sorted(nums) 함수는 정렬한 새로운 리스트를 만든다.

nums = [3, 5, 1, 4, 2]
new_nums = sorted(nums) # 정렬한 새로운 리스트를 만든다.
print(nums, new_nums) # [3, 5, 1, 4, 2] [1, 2, 3, 4, 5]

.reverse()

주어진 리스트를 반대로 뒤집는다. 정렬과는 관계가 없다.

 

 

 

리스트에서 인덱스 위치에 해당하는 원소를 추출한 후 반환한다.