리스트

a = [1,2,3,4,5]
a = list()
a = []
a = [0] * 10

array = [i for i in range(1,10) if i % 2 == 1]
array = [i*i for i in range(1,10)

연산

a/b # 그냥 나눈 값
a//b # 몫
a%b # 나머지
a ** b # 거듭연산자

a=[1,2,3,4,5]
b=a[1:4] # 2 3 4
# 마지막 원소 전까지

리스트 컴프리헨션

2차원 배열 초기화

n=3
m=4
array = [[0] * m for _ in range(n)]
# [0] * m 배열의 방을 n 만큼 만들어라 
# N * M
# 순서 반대인 것에 유의

잘못된 방법

n = 3
m = 4
array = [[0]*m] * n

array[1][1]=5
print(array)

리스트 메서드

a = [1,2,3,3,3,4,5]

a.sort()
a.sort(reverse=True)
a.reverse()

a.insert(2,300)
# 2번째에 300 숫자 추가

a.add(6)
# 하나만 가능

a.update([10,11,12])
# 여러개 가능

a.count(3)
# 3숫자 몇개있는지 세기

a.remove(3)
# 하나만 사라짐

유의점

insert() / remove()

등은 시간복잡도가 O(n)

특정한 값을 모두 제거하려면?

a=[1,2,3,4,3,3,3,5,5,5,5,5]
delete_list = [3,5]

result = [i for i in a if i not in delete_list]

문자열

a = "\"python\""
print(a)
# "python"

a = "hi"
print(a*5)
# hihihihihi

a = "abcde"
print(a[2:4])
# cd

튜플

• 튜플은 한번 선언된 값을 변경할 수 없다

• 리스트는 [] // 튜플은 ()

사전자료형

key - value ( map )

data = dict() # {} 와 같음 
data['1'] = 2
data['얼수'] = '안녕'

print(data)

사전자료형 출력

• dictionary() - python에서의 map은 다른 용도임

• 일반 map = dict()

data = dict()

data['a']=1
data['b']=2
data['c']=3

key_set = data.keys()
value_set = data.values()

print(key_set)
print(value_set)

for i in key_set:
    print(i)

set

• {} 이렇게 선언해도 set임

data = {1,2,1,1,1,1,1,12}

print(data)

data = set([1,2,3,4,1,1,1,1])

print(data)

합 & 교 & 차

a = { 1,2,3,4,5}

b= {3,4,5,6,7}

print(a|b)
print(a&b)
print(a-b)

• | or // & and 도 하나만 사용한다

조건문

x=13

if x>15:
    print("hi")
elif x==15:
    print("bye")
else:
    print("aa")

들여쓰기

• 들여쓰기가 같으면 함께 실행됨

스페이스바 4번이 표준 ( tab 도 상관없음 )
스페이스바로 하는 게 권장

논리연산자

print(10==10 or 5==4)

and or 연산을 그냥 영어로 한다

• pass = 아무것도 처리하고 싶지 않을 때

조건부 표현식

score = 85

result = "success" if score>=80 else 'fail'

print(result)

응용

원래 코드

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

result=[]

for i in a:
    if i not in remove_set:
        result.append(i)

print(result)

줄인 코드

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

result=[i for i in a if i not in remove_set]

print(result)

for

range(1,10) 
# (for i = 1; i<10; i++)

range(5)
# 첫 값이 0으로 자동설정

람다식

print((lambda a,b: a+b)(3,7))

입출력

n=int(input()) # 줄바꿈 함수
a = list(map(int,input().split()))

print(a)

• input()의 경우 속도가 매우 느림

import sys
data = sys.stdin.readline().rstrip()
print(data)

• 그래서 sys를 이용

• rstrip() 은 줄바꿈을 제거해주는 함수

print()

• print = println

print("안녕" + str(7) + "입니다")

answer = 88

print(f"안녕 {7}")
print(f"안녕 {answer}")

• "" 앞에 f 붙이고 {} 로 표현할 수도 있다.

라이브러리

• 내장함수 ( print, input, sorted )

• itertools : 순열과 조합등

• heapq : heap

• bisect : 이진탐색

• collections : 덱, 카운터 등

• math

print(sum([1,2,3,4]))
print(min([1,2,3,4]))
print(max([1,2,3,4]))

result = eval("(3+5)*7")

print(result)

객체정렬

result = sorted([('a',30),('b',60),('c',70)],key=lambda x:x[1],reverse=False)

print(result)

itertools

• permutations

• combinations

permutations

from itertools import permutations

data = ['a','b','c']

result = list(permutations(data,2))
print(result)

combinations

from itertools import combinations

data = ['a','b','c']

result = list(combinations(data,2))
print(result)

product

순열 중복허용

from itertools import product

data = ['a','b','c']

result = list(product(data,repeat=2))
print(result)

combinations_with_replacement

조합 중복허용

from itertools import combinations_with_replacement

data = ['a','b','c']

result = list(combinations_with_replacement(data,2))
print(result)

heapq

• priorityQueue 를 사용할 수도 있지만 heapq가 더 빠르다

파이썬 heap = 최소힙 O(nlogn)

heapq.heappush()
# 삽입

heapq.heappop()
# 제거

사용법

def heapsort(data):
    h=[]
    result=[]

    for value in data:
        heapq.heappush(h, value)

    for i in range(len(h)):
        result.append(heapq.heappop(h))
    return result

result = heapsort([1,3,5,7,9,11,131,15])

print(result)

• 최소힙만 제공하고 최대힙은 제공하지 않는다. 그래서 음수를 붙여서 넣고, 음수를 붙여서 꺼내면 최대힙으로 정렬된다

def heapsort(data):
    h=[]
    result=[]

    for value in value:
        heapq.heappush(h, -data)

    for i in range(len(h)):
        result.append(-heapq.heappop(h))
    return result

result = heapsort([1,3,5,7,9,11,131,15])

print(result)

bisect

• 이진탐색

bisect_left()
bisect_right()

from bisect import bisect_left,bisect_right

a=[1,2,4,4,4,4,5,8,9]

x=4

print(bisect_left(a,x))
# 2
print(bisect_right(a,x))
# 6

어느 곳에 삽입될지 정하는 method

만약 값이 동일하다면 left는 왼쪽, right는 오른쪽 index를 반환해줌

개수 세기

• O(logN) 시간 보장

from bisect import bisect_left,bisect_right

def count_by_range(a,left_value,right_value):
    right_index = bisect_right(a,right_value)
    left_index = bisect_left(a,left_value)
    return right_index-left_index

a=[1,2,3,3,3,3,3,4,4,4,4,8,9]

print(count_by_range(a,4,4))
# 값이 4인 data 개수 출력
print(count_by_range(a,-1,3))
# 값이 -1~3인 데이터 개수 출력

Collections

deque

• 스택과 큐를 모두 포함

from collections import deque

data = deque([1,2,3,4])

data.popleft()

data.append(33)

print(data)

queue 를 사용할때는

poll() = popleft()
offer() = append()

Counter

from collections import Counter

counter = Counter(['blue','blue','red'])

print(counter['blue'])

횟수 세줌

math

import math

print(math.factorial(10))

print(math.sqrt(7))
# 제곱근

print(math.gcd(21,14))
# 최대공약수