Python NumPyのchar.endswith()で配列内の文字列要素が指定した接尾辞で終わるか判定する方法

配列内の文字列要素が指定した接尾辞（サフィックス）で終わる場合に True となるブール配列（真偽値の配列）を返すには、Python NumPy の numpy.char.endswith() メソッドを使用します。

第1引数には入力配列を、第2引数には判定したい接尾辞を指定します。numpy.char モジュールは、numpy.str_ 型の配列に対してベクトル化された文字列操作を提供するモジュールです。Python の組み込みメソッド str.endswith() を配列全体に一括適用できるため、ループ処理を書く必要がなく、効率的に判定を行えます。

実行手順

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

import numpy as np

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

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

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

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)

文字列要素が接尾辞で終わる位置で True となるブール配列を返すには、numpy.char.endswith() メソッドを使用します。

print("\nResult (endswith)...\n", np.char.endswith(arr, 'E'))

完全なコード例

import numpy as np

# 文字列の1次元配列を作成
arr = np.array(['KATIE', 'JOHN', 'KATE', 'AmY', 'BRADley'])

# 配列を表示
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() を使用して、
# 各要素が接尾辞 'E' で終わるかどうかを判定
# 第1引数：入力配列、第2引数：接尾辞
print("\nResult (endswith)...\n", np.char.endswith(arr, 'E'))

出力結果

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

Array datatype...
<U7

Array Dimensions...
1

Our Array Shape...
(5,)

Number of elements in the Array...
5

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

補足ポイント

上記の出力からわかるように、'KATIE' と 'KATE' は末尾が 'E' であるため True となり、それ以外の要素は False と判定されています。

なお、np.char.endswith() では大文字と小文字が区別される点に注意してください。例えば 'AmY' の末尾は 'Y' であり 'E' ではないため False になります。大文字小文字を区別せずに判定したい場合は、あらかじめ np.char.upper() や np.char.lower() で配列の要素を正規化してから判定するとよいでしょう。