[파이썬][Numpy] 배열 정렬과 검색 예제

24 Aug 2023

numpy

Numpy 배열에서의 정렬과 검색을 수행하는 예제를 제공하겠습니다. Numpy는 배열을 정렬하고 특정 조건에 따라 배열을 검색하는 데 사용할 수 있는 다양한 함수를 제공합니다.

배열 정렬 예제:

import numpy as np

# 배열 생성 예제
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])

# 배열 정렬 (오름차순)
sorted_arr = np.sort(arr)
print("오름차순 정렬:")
print(sorted_arr)

# 배열 정렬 (내림차순)
reverse_sorted_arr = np.sort(arr)[::-1]
print("\n내림차순 정렬:")
print(reverse_sorted_arr)

# 원본 배열 정렬 (in-place 정렬)
arr.sort()
print("\n원본 배열 in-place 정렬:")
print(arr)

배열 검색 예제:

# 배열 생성 예제
arr = np.array([3, 1, 4, 1, 5, 9, 2, 6, 5, 3, 5])

# 고유한(unique) 값 찾기
unique_values = np.unique(arr)
print("고유한 값:", unique_values)

# 조건에 맞는 원소 찾기
condition = arr > 3
filtered_arr = arr[condition]
print("\n조건에 맞는 원소:", filtered_arr)

위의 코드에서는 다음과 같은 작업을 수행합니다:

배열 정렬 예제: 배열을 오름차순과 내림차순으로 정렬하며, sort() 메서드를 사용하여 원본 배열을 in-place로 정렬합니다.
배열 검색 예제: 배열에서 고유한(unique) 값을 찾는 방법과 조건에 맞는 원소를 검색하는 방법을 보여줍니다.

실행 결과는 다음과 같을 것입니다:

배열 정렬 결과:

오름차순 정렬:
[1 1 2 3 3 4 5 5 5 6 9]

내림차순 정렬:
[9 6 5 5 5 4 3 3 2 1 1]

원본 배열 in-place 정렬:
[1 1 2 3 3 4 5 5 5 6 9]

배열 검색 결과:

고유한 값: [1 2 3 4 5 6 9]

조건에 맞는 원소: [4 5 9 6 5 5]

Numpy를 사용하면 배열을 효과적으로 정렬하고 원하는 조건에 따라 검색할 수 있습니다.