Numpy를 사용하여 데이터 필터링과 선택하는 예제를 제공하겠습니다. 데이터 필터링은 배열에서 특정 조건을 충족하는 원소만 선택하는 것을 의미하며, 배열의 일부를 선택하는 방법을 보여줍니다.

import numpy as np

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

# 조건을 충족하는 원소 선택
condition = (data > 3)  # 3보다 큰 원소 선택
filtered_data = data[condition]
print("3보다 큰 원소 선택:")
print(filtered_data)

# 특정 행 선택
selected_row = data[1]  # 두 번째 행 선택
print("\n두 번째 행 선택:")
print(selected_row)

# 특정 열 선택
selected_column = data[:, 2]  # 세 번째 열 선택
print("\n세 번째 열 선택:")
print(selected_column)

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

• 조건을 충족하는 원소 선택: data 배열에서 조건에 맞는 원소만 선택합니다. 이를 위해 불리언 배열을 생성하고 이를 사용하여 데이터를 필터링합니다.

• 특정 행 선택: 특정 행을 선택하는 방법을 보여줍니다. 행은 배열의 첫 번째 차원을 따라 선택됩니다.

• 특정 열 선택: 특정 열을 선택하는 방법을 보여줍니다. 열은 배열의 두 번째 차원을 따라 선택됩니다.

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

3보다 큰 원소 선택:
[4 5 6 7 8 9]

두 번째 행 선택:
[4 5 6]

세 번째 열 선택:
[3 6 9]

Numpy를 사용하면 배열에서 원하는 데이터를 필터링하고 선택하는 데 유용한 다양한 방법을 활용할 수 있습니다.