Computer >> 컴퓨터 >  >> 프로그래밍 >> Python

Python NumPy char.startswith() 활용법 – 문자열이 특정 접두사로 시작하는지 확인하는 부울 배열 반환하기

배열 내 문자열 요소가 주어진 접두사(prefix)로 시작하는 위치에서 True를 갖는 부울(boolean) 배열을 반환하려면 Python NumPy의 numpy.char.startswith() 메서드를 사용합니다. 이 메서드는 bool 값으로 구성된 배열을 출력합니다.

주요 매개변수는 다음과 같습니다.

  • 첫 번째 매개변수: 검사 대상인 입력 배열
  • 두 번째 매개변수: 확인할 접두사(prefix)
  • start(선택 사항): 지정한 위치부터 검사를 시작합니다.
  • end(선택 사항): 지정한 위치에서 비교를 중단합니다.

구현 단계

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

import numpy as np

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

arr = np.array(['KATIE', 'JOHN', 'KATE', 'KmY', 'BRAD'])

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

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.startswith() 메서드를 사용하여, 배열의 문자열 요소가 주어진 접두사로 시작하는 위치에서 True를 반환하는 부울 배열을 얻습니다.

print("\nResult (startswith)...\n",np.char.startswith(arr, 'K', start = 0, end = 2))

전체 예제 코드

import numpy as np

# 문자열로 구성된 1차원 배열 생성
arr = np.array(['KATIE', 'JOHN', 'KATE', 'KmY', 'BRAD'])

# 배열 출력
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.startswith() 메서드를 사용하여
# 문자열 요소가 주어진 접두사로 시작하는지 검사하고 부울 배열 반환
# 이 메서드는 bool 값으로 구성된 배열을 출력합니다.
print("\nResult (startswith)...\n",np.char.startswith(arr, 'K', start = 0, end = 2))

실행 결과

Array...
['KATIE' 'JOHN' 'KATE' 'KmY' 'BRAD']

Array datatype...
<U5

Array Dimensions...
1

Our Array Shape...
(5,)

Number of elements in the Array...
5

Result (startswith)...
[ True False True True False]

위 실행 결과에서 볼 수 있듯이, start=0, end=2 조건으로 첫 두 글자만 검사했기 때문에 'KATIE', 'KATE', 'KmY'는 'K'로 시작하므로 True가 되고, 'JOHN'과 'BRAD'는 False가 반환됩니다.