[NumPy] NumPy로 배열을 생성하는 방법

NumPy는 배열을 다양한 방법으로 쉽게 생성할 수 있는데, 그 중에서도 자주 사용되는 함수들이 있다.

 

1. np.arange()

np.arange()는 일정 간격의 값을 가지는 배열을 생성할 수 있다. 기본적으로는 0부터 시작해서 지정한 숫자까지 값을 생성하는데, 원하는 간격으로 숫자를 설정할 수도 있다.

import numpy as np

# 기본 설정: 0부터 10까지 (10은 포함되지 않음), 1씩 증가
array1 = np.arange(0, 10)
print(array1)  # [0 1 2 3 4 5 6 7 8 9]

# 사용자 지정 증가 폭
array2 = np.arange(0, 10, 2)
print(array2)  # [0 2 4 6 8]

 

• 첫 번째 예시는 0부터 10까지 (10은 포함되지 않음) 1씩 증가하는 배열을 만든다.
• 두 번째 예시는 2씩 증가하는 배열을 만든다. 간격을 원하는 대로 설정할 수 있다.

2. np.zeros()와 np.ones()

np.zeros()와 np.ones()는 각각 0과 1로 채워진 배열을 생성하는 함수야. 이 배열들은 초기값을 설정하거나, 빈 데이터를 생성할 때 유용하게 사용돼.

# 3x4 형태의 0으로 채워진 배열 생성
zeros_array = np.zeros((3, 4))
print(zeros_array)

# 2x5 형태의 1로 채워진 배열 생성
ones_array = np.ones((2, 5))
print(ones_array)

• 첫 번째 예시는 3x4 형태로 0으로 채워진 배열을 만든다.
• 두 번째 예시는 2x5 형태로 1로 채워진 배열을 만든다.

3. 정리 및 요약

함수설명예시 코드출력 예시

함수	설명	예시	출력 예시
np.arange()	일정 간격의 값을 가지는 배열을 생성 (시작, 종료, 간격 지정 가능)	np.arange(0, 10) np.arange(0, 10, 2)	[0 1 2 3 4 5 6 7 8 9] [0 2 4 6 8]
np.zeros()	0으로 채워진 배열 생성 (모양은 튜플로 지정)	np.zeros((3, 4))	[[0. 0. 0. 0.] [0. 0. 0. 0.] [0. 0. 0. 0.]]
np.ones()	1로 채워진 배열 생성 (모양은 튜플로 지정)	np.ones((2, 5))	[[1. 1. 1. 1. 1.] [1. 1. 1. 1. 1.]]