함수 / *args, **kwargs / generator, yield

01 Oct 2021 - 1 minute read

함수

---

• 함수란? 입력값을받아서출력값을내어놓는것

  • 입력 => [함수] => 출력

• 기본형태

    def 함수명(매개변수):
        수행할 문장1
        수행할 문장2
        ...
  

*args / **kwargs

---

• 매개변수(parameter) / 인수(arguments)

  • 매개변수 : 함수에 입력으로 전달된 값을 받는 변수

  • 인수 : 함수를 호출할 때 전달하는 입력값

    def add(a, b)   # 매개변수
        return a + b
  
    print(add(1, 2))    # 인수
  

• *args : 입력값이 몇 개가 될 지 모를 때

    def 함수명(*매개변수){
        수행할 문장
        ...
    }
  

  • 인수는 튜플 형태로 변환, 매개변수에 저장됨

      def test(*args):
          for i in args:
              print(i)
      test(1, 2, 3, 4)
    
      ### 출력결과
      ### 1
      ### 2
      ### 3
      ### 4
    

    
    

      def add_sub(choice, *args):
          if choice == "add":
              res = 0
              for i in args:
                  res = res + i
          elif choice == "sub":
              res = 1
              for i in args:
                  res = res - i
          return res
    
      res = add_sub('add', 1, 2, 3, 4, 5)
      print(res)
      res = add_sub('sub', 10, 1)
      print(res)
    
      ### 출력결과
      ### 15
      ### -10
    

    
    

      a, *b = [1, 2, 3, 4]
      print(a)
      print(b)
      ### 출력결과
      ### 1
      ### [2, 3, 4]
    
    
    
      def test(a, *b):
          return a, b
    
      print(test(1, 2, 3, 4))
      ### 출력결과
      ### (1, (2, 3, 4))
    

• **kwargs

  • 인수는 딕셔너리 형태로 변환, 매개변수에 저장됨

  • (키워드 = 값) 형태로만 함수 호출 가능

      def info(**kwargs):
          print(kwargs)
    
      info(name="이감자", age="20")
    
      ### 출력결과
      ### {'name': '이감자', 'age': '20'}
    

매개변수에 초깃값 설정하기

---

• def 함수명(매개변수=값):
      수행할 문장
  

• 초기화시키고 싶은 매개변수는 항상 맨 뒤에 위치해야 한다. 그렇지 않으면 에러 발생

    def add(a, b=10):
        print(a+b)
  
    add(5)
    ### 출력 > 15
    add(5, 5)
    ### 출력 > 10
  
  
    ### def add(a=10, b):
    ###    print(a+b)
    ### > 에러!
  

generator / yield

---

• generator : 메모리를 효율적으로 사용하면서 반복을 수행하도록 돕는 객체

• yield : 원하는 값 리턴, 실행 흐름을 일시 정지하여 함수를 재활용할 수 있는 상태로 만듦

    def gen(n):
        while n>0:
            yield n
            n=n-1
  
    print(gen(10))
  
    ### 출력결과
    ### <generator object gen at 0x000001CED6C637B0>
    ### => generator 객체 반환
  
    print(list(gen(10)))
  
    ### 출력결과
    ### [10, 9, 8, 7, 6, 5, 4, 3, 2, 1]
  

  
  

    def gen(nums):
        for num in nums:
            yield num * num
  
    nums = [i for i in range (6, 11)]
    res = gen(nums)
    for num in res:
        print(num)
  
    ### 출력결과
    ### 36
    ### 49
    ### 64
    ### 81
    ### 100