numpy 배열 생성하기 zeros(), ones()

numpy에서 자주쓰이는 zeros(), ones() 함수를 정리합니다.

 

zeros() 함수

import numpy as np
shape = (3,2) #row=3, column=2
zero_array = np.zeros(shape)

np.zeros() 함수는 위와 같이 크기를 인자값으로 받게되는데 행렬의 크기나 ndarray의 크기를 지정해주면 됩니다.

 

이렇게 만든  zero_array를 출력해보면

print(zero_array)
print(type(zero_array))
print(zero_array.dtype)

- zero_array의 타입은 numpy.ndarray 타입으로 넘파이 배열입니다.

- zero_array의 데이터의 타입은 'float64' 입니다.

※ zeros()함수의 default dtype = 'float64'이기 때문입니다.

 

다음은 ones() 함수 입니다.

ones() 함수

import numpy as np
shape=(3,2)
one_array = np.ones(shape, dtype='int64')

생성방법은 위와 마찬가지로 생성할 수 있습니다. 여기서는 dtype='int64'를 적용하여 데이터타입을 바꾸었습니다.

print(one_array)
print(type(one_array))
print(one_array.dtype)

- one_array의 타입은 numpy.ndarray 타입으로 넘파이 배열입니다.

- one_array의 데이터의 타입은 변경한 'int64' 입니다.