Python NumPyで配列内の文字列が指定した接尾辞で終わるかを判定し、ブール配列を返す方法

配列内の文字列要素が指定した接尾辞（サフィックス）で終わる位置にTrueが格納されたブール配列を返すには、Python NumPyのnumpy.char.endswith()メソッドを使用します。このメソッドはbool型の配列を出力します。第1引数には入力配列を、第2引数には接尾辞を指定します。さらに、オプションのstart引数を指定するとその位置から判定を開始し、オプションのend引数を指定するとその位置で比較を終了します。

手順

まず、必要なライブラリをインポートします。

import numpy as np

文字列の1次元配列を作成します。

arr = np.array(['KATIE', 'JOHN', 'KATE', 'AmY', 'CRADlE'])

作成した配列を表示します。

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

配列のデータ型を確認します。

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

配列の次元数を取得します。

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

配列の形状を取得します。

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

配列の要素数を取得します。

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

配列内の文字列要素が接尾辞で終わるかどうかを判定したブール配列を返すには、numpy.char.endswith()メソッドを使用します。

print("\nResult (endswith)...\n",np.char.endswith(arr, 'E', start = 2, end = 4))

コード例

import numpy as np

# Create a One-Dimensional array of strings
arr = np.array(['KATIE', 'JOHN', 'KATE', 'AmY', 'CRADlE'])

# Displaying our array
print("Array...\n",arr)

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

# Get the dimensions of the Array
print("\nArray Dimensions...\n",arr.ndim)

# Get the shape of the Array
print("\nOur Array Shape...\n",arr.shape)

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

# To return a boolean array which is True where the string element in array ends with suffix, use the numpy.char.endswith() method in Python Numpy
# The method outputs an array of bools.

print("\nResult (endswith)...\n",np.char.endswith(arr, 'E', start = 2, end = 4))

出力結果

Array...
['KATIE' 'JOHN' 'KATE' 'AmY' 'CRADlE']

Array datatype...
<U6

Array Dimensions...
1

Our Array Shape...
(5,)

Number of elements in the Array...
5

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