특정 열을 기준으로 NumPy 배열 요소를 어떻게 그룹화할 수 있나요?

NumPy의 함수별 그룹화

NumPy는 요소 그룹화 작업을 포함하여 배열 조작을 위한 여러 기능을 제공합니다. 그러한 작업 중 하나는 지정된 키를 기반으로 배열의 요소를 그룹화할 수 있는 groupby입니다.

특정 문제

다음 배열 a를 고려하세요.

a = array([[ 1, 275],
       [ 1, 441],
       [ 1, 494],
       [ 1, 593],
       [ 2, 679],
       [ 2, 533],
       [ 2, 686],
       [ 3, 559],
       [ 3, 219],
       [ 3, 455],
       [ 4, 605],
       [ 4, 468],
       [ 4, 692],
       [ 4, 613]])

첫 번째 열을 기준으로 요소를 그룹화한다고 가정해 보겠습니다. 이 경우 출력은 다음과 같습니다.

array([[[275, 441, 494, 593]],
       [[679, 533, 686]],
       [[559, 219, 455]],
       [[605, 468, 692, 613]]], dtype=object)

Solution

NumPy에는 직접적인 그룹별 기능이 없지만 이를 달성하는 것이 가능합니다. 다음 접근 방식을 사용합니다.

# Sort the array by the first column
a = a[a[:, 0].argsort()]

# Find the unique values in the first column as keys
keys = np.unique(a[:, 0])

# Create an array to hold the grouped elements
grouped = []

# Iterate through the keys
for key in keys:
    # Create a mask to select elements with the given key
    mask = (a[:, 0] == key)
    
    # Append the selected elements to the grouped array
    grouped.append(a[mask][:, 1])

이 솔루션은 첫 번째 열을 기준으로 배열의 요소를 효율적으로 그룹화합니다. 명시적으로 groupby 기능을 사용하지 않더라도 말이죠.

위 내용은 특정 열을 기준으로 NumPy 배열 요소를 어떻게 그룹화할 수 있나요?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!