Python numpy : ones (1로 채워진 array 생성)

 

 

 

 

numpy의 ones() method는 1로만 채워진 array를 생성합니다.

 

아래 예시들을 봅시다.

 

import numpy as np

arr_ones = np.ones(shape=3)
print(arr_ones)


-- Result
[1. 1. 1.]

ones(3)의 의미는

3개의 1을 요소로서 가지는 1차원 array를 생성하라는 의미입니다.

이렇게 ones() method는 기본적으로 1차원 array를 생성합니다.

 

 

 

 

import numpy as np

arr_ones = np.ones(shape=3, dtype=int)
print(arr_ones)


-- Result
[1 1 1]

위 예시처럼 dtype 옵션을 이용해서 data type을 명시해줄 수 있습니다.

위 예시는 1의 data type을 모두 정수(int, integer)로 설정했기 때문에 소수점이 사라졌습니다.

 

 

 

 

 

import numpy as np

arr_ones = np.ones(shape=(2, 3))
print(arr_ones)


-- Result
[[1. 1. 1.]
 [1. 1. 1.]]

ones() method에 숫자가 아니라 tuple 형태의 값을 입력해주면 입력된 형태를 가지는 다차원 array를 생성해줍니다.

위 예시는 tuple의 길이가 2이므로 2차원 array가 생성되며

2개의 행, 3개의 열을 가진 array가 생성되었습니다.

 

 

 

 

 

import numpy as np

arr_ones = np.ones(shape=(2, 3, 5))
print(arr_ones)


-- Result
[[[1. 1. 1. 1. 1.]
  [1. 1. 1. 1. 1.]
  [1. 1. 1. 1. 1.]]

 [[1. 1. 1. 1. 1.]
  [1. 1. 1. 1. 1.]
  [1. 1. 1. 1. 1.]]]

이렇게 3차원 array도 생성할 수 있습니다.

 

결과를 보면 행 개수 = 2, 열 개수 = 3인 array를 생성하는데

이 array에 있는 각각의 요소의 개수가 5개인 것을 알 수 있습니다.

 

 

[
    [ [1. 1. 1. 1. 1.] [1. 1. 1. 1. 1.] [1. 1. 1. 1. 1.] ]
    [ [1. 1. 1. 1. 1.] [1. 1. 1. 1. 1.] [1. 1. 1. 1. 1.] ]
]

 

결과 array를 좀 더 직관적으로 보기 쉽게 배열하면 위와 같습니다.

행 개수 = 2개, 열 개수 = 3개이며 각각의 요소는 총 5개 짜리 array가 되는 것이죠.