[Numpy] np.ones()

Numpy np.ones() 함수

np.ones() 함수는 지정된 shape의 배열을 생성하고 모든 요소를 1로 채웁니다.

문법

numpy.ones(shape, dtype=None, order='C')

매개변수

• shape : 배열의 형태를 지정하는 정수 또는 정수의 튜플.
• dtype : 배열 요소의 데이터 형식. 선택적 매개변수이며, 기본값은 numpy.float64입니다.
• order : 배열의 요소를 저장하는 순서를 지정합니다. 선택적 매개변수이며, C(행 기준) 또는 F(열 기준) 중 하나를 사용할 수 있습니다. 기본값은 C입니다.

반환값

주어진 shape과 dtype을 가지는 배열. 모든 요소는 1로 채워집니다.

사용 예시

import numpy as np

# 1차원 배열 생성
arr1 = np.ones(5)
print(arr1)

# 2차원 배열 생성
arr2 = np.ones((2, 3))
print(arr2)

# 데이터 타입 변경
arr3 = np.ones((2, 2), dtype=int)
print(arr3)

출력 결과

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

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

[[1 1]
 [1 1]]