NumPy ValueError: concatenation of incompatible shapes 오류 해결하기

소개

NumPy를 사용할 때 'ValueError: concatenation of incompatible shapes' 오류는 자주 발생하는 문제 중 하나입니다. 이 오류는 주로 배열을 연결(concatenate)하려고 할 때 두 배열의 모양(shape)이 서로 호환되지 않을 때 발생합니다. 오늘은 이 오류의 원인과 해결 방법을 함께 알아보겠습니다.

에러 발생 예시 코드

먼저, 'ValueError: concatenation of incompatible shapes' 오류가 발생할 수 있는 간단한 예시 코드를 보겠습니다.

import numpy as np

# 두 개의 배열 생성
arr1 = np.array([[1, 2, 3], [4, 5, 6]])
arr2 = np.array([[7, 8]])

# 배열을 수직으로 연결하려고 시도
result = np.concatenate((arr1, arr2), axis=0)
print(result)

에러 해결 방법

1. 배열 모양 확인

우선 연결하려는 두 배열의 모양을 확인하세요. 배열의 차원(dimension)과 크기(size)가 맞지 않으면 연결할 수 없습니다. 아래의 코드를 사용해 배열의 모양을 확인해 봅시다.

print("arr1 shape:", arr1.shape)
print("arr2 shape:", arr2.shape)

2. 배열을 호환되도록 수정

배열의 크기를 맞추기 위해 차원을 늘리거나, 필요에 따라 배열을 변경할 수 있습니다. 예를 들어, 아래와 같이 'arr2'의 형상을 조정하여 오류를 피할 수 있습니다.

import numpy as np

# 두 개의 배열 생성
arr1 = np.array([[1, 2, 3], [4, 5, 6]])
arr2 = np.array([[7, 8], [9, 10]])  # arr2의 형식을 변경

# 배열을 수직으로 연결
result = np.concatenate((arr1, arr2), axis=0)
print(result)

3. 연결할 축(axis) 재확인

연결하려는 축이 올바른지 확인하세요. 기본적으로 수직(axis=0) 또는 수평(axis=1)으로 연결할 수 있습니다. 연결하고자 하는 방향에 맞게 배열을 구성해야 합니다.

import numpy as np

# 두 개의 배열 생성
arr1 = np.array([[1, 2, 3]])
arr2 = np.array([[4, 5, 6]])

# 배열을 수평으로 연결
result = np.concatenate((arr1, arr2), axis=1)
print(result)

마무리

오늘은 NumPy에서 'ValueError: concatenation of incompatible shapes' 오류가 발생했을 때의 원인과 해결 방법에 대해 알아보았습니다. 배열의 모양을 확인하고, 배열을 조정하거나 연결 축을 재확인함으로써 이러한 오류를 극복할 수 있습니다. NumPy를 사용할 때는 항상 배열의 형상을 주의 깊게 살피는 것이 중요하다는 점을 기억하세요!