본문 바로가기

PYTHON

[python] 기초문법 - 함수만들기

파이썬 기초문법 for 코딩테스트

함수만들기

 

두 매개변수를 받아서 더해서 결과를 리턴하는 함수 만드는 방법

def add(a,b):#함수 정의

    c=a+b

    print(c)

#main 스크립트

add(3,2)#함수호출, 5출력

add(5,7)#12출력

#함수는 main 스크립트 위에 작성해야한다. 

#add를 먼저 호출하고 그 밑에 함수가 정의되어있으면, 컴퓨터는 add를 호출하는 시점에 함수가 정의되어있는지 모르기 때문이다.

 

#함수 리턴 값 받기

def add(a,b):#함수 정의

    c=a+b

    return c 

#main 스크립트

print(add(3,2))#5를 리턴 받음. 5출력

#리턴하는 순간 함수호출이 종료 된다.

 

def add(a, b):

    c=a+b

    d=a-b

    return c,d #여러개 의 값 리턴 가능 c++에서는 불가능한 방법

print(add(3,2))#(5,1) 출력, 튜플형태로 리턴된다.

 

#소수만 리턴하는 함수

def isPrime(x):

    for i in range(2,x):

        if x%i==0:

            return False

        return True

#main 스크립트

a=[12,13,7,9,19]

for y in a:

    if isPrime(y):

        print(y, end=' ')#소수라면 출력

#출력결과: 13,  7, 19 출력(소수만)

 

함수 안에서 변수와 main 스크립트 안에서 변수명은 같아도 된다. 함수안에서 선언된 변수는 지역변수 이기때문에 main 스크립트와 다른 변수이다.