Python에서 3D 배열을 2D 배열로 변환

이 자습서에서는 Python에서 3D 배열을 2D 배열로 변환하는 방법에 대해 설명합니다.

Python에서numpy.reshape()함수를 사용하여 3D 배열을 2D 배열로 변환

[numpy.reshape()함수] (numpy.reshape - NumPy v1.20 매뉴얼) 모양을 변경합니다. 데이터를 변경하지 않고 배열의. numpy.reshape()는 지정된 차원의 배열을 반환합니다. 예를 들어, 차원이(4, 2, 2)인 3D 배열이 있고이를(4, 4)차원의 2D 배열로 변환하려는 경우.

다음 코드 예제는numpy.reshape()함수를 사용하여(4, 2, 2)차원의 3D 배열을 Python에서(4, 4)차원의 2D 배열로 변환하는 방법을 보여줍니다.

import numpy

arr = numpy.array(
    [[[0, 1], [2, 3]], [[4, 5], [6, 7]], [[8, 9], [10, 11]], [[12, 13], [14, 15]]]
)
newarr = arr.reshape(4, 2 * 2)
print(newarr)

출력:

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]

위의 코드에서 먼저numpy.array()함수를 사용하여 3D 배열arr를 초기화 한 다음numpy.reshape()함수를 사용하여 2D 배열newarr로 변환합니다.

다음 코드 예제는 어떤 이유로 3D 배열의 정확한 치수를 알지 못하는 경우 동일한 작업을 수행하는 또 다른 방법을 보여줍니다.

import numpy

arr = numpy.array(
    [[[0, 1], [2, 3]], [[4, 5], [6, 7]], [[8, 9], [10, 11]], [[12, 13], [14, 15]]]
)
newarr = arr.reshape(arr.shape[0], (arr.shape[1] * arr.shape[2]))
print(newarr)

출력:

[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]
 [12 13 14 15]]

위 코드에서numpy.shape()함수를 사용하여newarr의 크기를 지정합니다. numpy.shape()함수는 개수를 포함하는 튜플을 반환합니다. 배열의 각 차원에있는 요소.