Python NumPy에서 문자열 배열의 요소별 길이 구하기: np.char.str_len() 활용법

Python의 NumPy 라이브러리에서 문자열 배열의 각 요소 길이를 요소별(element-wise)로 반환하려면 numpy.char.str_len() 메서드를 사용하면 됩니다. 이 메서드는 각 문자열의 길이를 담은 정수형 출력 배열을 반환합니다.

단계별 진행 과정

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

import numpy as np

문자열로 구성된 1차원 배열을 생성합니다.

arr = np.array(['Amy', 'Scarlett', 'Katie', 'Brad', 'Tom'])

생성한 배열을 화면에 출력합니다.

print("Array...\n",arr)

배열의 데이터 타입을 확인합니다.

print("\nArray datatype...\n",arr.dtype)

배열의 차원을 확인합니다.

print("\nArray Dimensions...\n",arr.ndim)

배열의 형태(shape)를 확인합니다.

print("\nOur Array Shape...\n",arr.shape)

배열에 포함된 요소의 개수를 확인합니다.

print("\nNumber of elements in the Array...\n",arr.size)

이제 numpy.char.str_len() 메서드를 사용해 문자열 배열의 각 요소 길이를 요소별로 반환합니다.

print("\nResult (length element-wise)...\n",np.char.str_len(arr))

전체 예제 코드

import numpy as np

# 문자열로 구성된 1차원 배열 생성
arr = np.array(['Amy', 'Scarlett', 'Katie', 'Brad', 'Tom'])

# 배열 출력
print("Array...\n",arr)

# 데이터 타입 확인
print("\nArray datatype...\n",arr.dtype)

# 배열의 차원 확인
print("\nArray Dimensions...\n",arr.ndim)

# 배열의 형태(shape) 확인
print("\nOur Array Shape...\n",arr.shape)

# 배열의 요소 개수 확인
print("\nNumber of elements in the Array...\n",arr.size)

# numpy.char.str_len() 메서드로 문자열 배열의 요소별 길이 반환
# 이 메서드는 정수형 출력 배열을 반환합니다.

print("\nResult (length element-wise)...\n",np.char.str_len(arr))

실행 결과

Array...
['Amy' 'Scarlett' 'Katie' 'Brad' 'Tom']

Array datatype...
<U8

Array Dimensions...
1

Our Array Shape...
(5,)

Number of elements in the Array...
5

Result (length element-wise)...
[3 8 5 4 3]

결과 해석

출력 결과를 보면 각 문자열의 길이가 순서대로 계산되었습니다. 'Amy'는 3글자, 'Scarlett'은 8글자, 'Katie'는 5글자, 'Brad'는 4글자, 'Tom'은 3글자이므로 결과 배열은 [3 8 5 4 3]이 됩니다. 참고로 데이터 타입 <U8은 유니코드 문자열을 의미하며, 배열 내 가장 긴 문자열('Scarlett')을 기준으로 최대 8글자까지 저장할 수 있음을 나타냅니다.