[넘파이 기초] 원하는 범위와 간격의 배열을 생성해보자, numpy.arange, numpy.linspace

원하는 범위의 배열을 생성해보자, numpy.arange¶

기존 범위를 생성하는 함수는 range() 함수였다. numpy에서는 arange 함수를 제공한다.

파라미터는 range 함수와 같이 np.arange(start,stop,step) 이다. stop(끝)부분의 숫자를 포함하지 않기에 주의가 필요하다.

In [2]:

import numpy as np

In [3]:

np.arange(10)

Out[3]:

array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])

In [4]:

np.arange(4, 9+1)

Out[4]:

array([4, 5, 6, 7, 8, 9])

In [5]:

np.arange(1, 13+1, 2)

Out[5]:

array([ 1,  3,  5,  7,  9, 11, 13])

끝함수에 무의식적으로 +1을 하다보면 아래와 같은 실수가 일어날 수 있으니 주의하자

In [6]:

np.arange(1, 10+1, 0.5)

Out[6]:

array([ 1. ,  1.5,  2. ,  2.5,  3. ,  3.5,  4. ,  4.5,  5. ,  5.5,  6. ,
        6.5,  7. ,  7.5,  8. ,  8.5,  9. ,  9.5, 10. , 10.5])

이럴때는 스템만큼의 숫자만 더해주는 방식이 유용하다.

In [8]:

np.arange(1, 10+0.5, 0.5)

Out[8]:

array([ 1. ,  1.5,  2. ,  2.5,  3. ,  3.5,  4. ,  4.5,  5. ,  5.5,  6. ,
        6.5,  7. ,  7.5,  8. ,  8.5,  9. ,  9.5, 10. ])

원하는 범위의 배열을 생성해보자, numpy.linspace¶

파라미터는 (시작, 끝, 개수)이다. 배열은 끝 숫자를 포함하며, 시작과 끝 사이에 원하는 개수만큼의 숫자를 반환한다.

In [7]:

np.linspace( 0, 25, 10 ) # 선형 공간 (시작, 끝, 개수)

Out[7]:

array([ 0.        ,  2.77777778,  5.55555556,  8.33333333, 11.11111111,
       13.88888889, 16.66666667, 19.44444444, 22.22222222, 25.        ])

끝의 숫자를 포함하고 싶지 않다면 endpoint = False 를 추가하면 된다.

In [9]:

np.linspace( 0, 25, 10, endpoint = False ) 

Out[9]:

array([ 0. ,  2.5,  5. ,  7.5, 10. , 12.5, 15. , 17.5, 20. , 22.5])