Python Pandas - Index의 데이터를 배열로 반환하는 방법

Pandas에서 Index 객체에 저장된 데이터를 배열 형태로 반환하려면 index.values 속성을 사용하면 됩니다. 이 속성은 Index 내부의 값을 NumPy 배열(ndarray) 형태로 돌려주기 때문에, 인덱스 데이터를 다른 연산이나 처리에 활용할 때 매우 유용합니다.

1. 라이브러리 임포트

가장 먼저 필요한 라이브러리를 임포트합니다.

import pandas as pd

2. Index 생성하기

문자열 요소들로 구성된 Pandas Index를 생성합니다.

index = pd.Index(['Car','Bike','Truck','Ship','Airplane'])

3. Index 출력하기

생성한 Index를 화면에 출력해 내용을 확인합니다.

print("Pandas Index...\n",index)

4. Index의 데이터를 배열로 반환하기

values 속성을 호출하면 Index의 데이터를 담고 있는 배열이 반환됩니다.

print("\nArray...\n",index.values)

전체 예제 코드

아래는 위 과정을 모두 포함한 전체 코드입니다. 참고로 Index의 전치(Transpose)를 확인하는 index.T 속성도 함께 살펴봅니다.

import pandas as pd

# 인덱스 생성
index = pd.Index(['Car','Bike','Truck','Ship','Airplane'])

# 인덱스 출력
print("Pandas Index...\n",index)

# Index의 데이터를 나타내는 배열 반환
print("\nArray...\n",index.values)

# 인덱스의 전치(Transpose) 출력
print("\nTranspose of the Pandas Index...\n",index.T)

실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

Pandas Index...
Index(['Car', 'Bike', 'Truck', 'Ship', 'Airplane'], dtype='object')

Array...
['Car' 'Bike' 'Truck' 'Ship' 'Airplane']

Transpose of the Pandas Index...
Index(['Car', 'Bike', 'Truck', 'Ship', 'Airplane'], dtype='object')

정리

index.values 속성을 사용하면 Pandas Index의 데이터를 손쉽게 NumPy 배열로 변환할 수 있습니다. 1차원 Index의 경우 전치(T)를 적용해도 동일한 결과가 반환되는 점도 확인할 수 있습니다. 이처럼 values 속성은 인덱스 데이터를 배열 기반 연산에 활용할 때 자주 사용되는 기본적인 방법입니다.