06. [Python] NumPy 기초 : 함수에서 배열 사용하는 방법

개요

NumPy에서 배열을 함수의 인자나 반환값(리턴값)으로 사용할 수 있습니다.

실습환경

NumPy는 아나콘다에 포함되어 있으므로 Jupyter Notebook에서 import만 해주면 됩니다.
아나콘다 설치 및 사용방법은 아래 링크를 확인해 주세요.
https://iot-lab.tistory.com/113

함수의 인자, 함수의 반환값으로 배열 사용하는 방법

아래 예제 함수 myFunc는 인수로 배열은 전달받고, 결과를 배열로 반환합니다.

[In]

import numpy as np

def myFunc(x):
    y = x * 2 + 1
    return y

a = np.array([[0, 1, 2], [3, 4, 5]])    # 2차원 배열
b = myFunc(a)   # 인자값으로 배열을 넘겨주고, 반환값으로 배열을 받음

print(b)

[결과]

다음에는 NumPy의 마지막 내용으로 NumPy의 추가적인 기능에 대해서 알아보도록 하겠습니다.