Python NumPyのrindex()メソッドで部分文字列が見つかる最大インデックスを返す方法

Python NumPyのnumpy.char.rindex()メソッドを使用すると、文字列の中で指定した部分文字列（sub）が見つかった位置のうち、最も大きい（右端の）インデックスを返すことができます。このメソッドはint型の要素を持つ出力配列を返し、部分文字列が見つからない場合はValueErrorが発生します。

引数の構成は以下のとおりです。

• 第1引数： 検索対象となる入力配列
• 第2引数： 検索する部分文字列

なお、rindex()は文字列の右側から検索を行うため、同じ部分文字列が複数回出現する場合でも、その中で最も後ろ側（インデックスが最大）の位置が返される点が特徴です。

手順

1. 必要なライブラリをインポートする

まず、NumPyライブラリをインポートします。

import numpy as np

2. 1次元の文字列配列を作成する

次に、文字列要素を含む1次元配列を作成します。

arr = np.array(['KATIE', 'KATE', 'BRATleuATp'])

3. 配列の情報を確認する

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

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)

4. rindex()メソッドで最大インデックスを取得する

numpy.char.rindex()メソッドを使用して、各文字列内で部分文字列「AT」が見つかった最も大きいインデックスを返します。

print("\nResult (rindex() method)...\n", np.char.rindex(arr, 'AT'))

完全なサンプルコード

import numpy as np

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

# 配列を表示
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)

# rindex()メソッドで部分文字列'AT'が見つかる最大インデックスを返す
print("\nResult (rindex() method)...\n", np.char.rindex(arr, 'AT'))

実行結果

Array...
['KATIE' 'KATE' 'BRATleuATp']

Array datatype...
<U10

Array Dimensions...
1

Our Array Shape...
(3,)

Number of elements in the Array...
3

Result (rindex() method)...
[1 1 7]

実行結果の解説

出力結果 [1 1 7] の意味は以下のとおりです。

• 'KATIE': 「AT」はインデックス1にのみ出現するため、1が返されます。
• 'KATE': 同様に「AT」はインデックス1にあるため、1が返されます。
• 'BRATleuATp': 「AT」はインデックス2と7の2箇所に出現しますが、rindex()は右側から検索するため、より大きい7が返されます。

このように、rindex()メソッドを使うことで、各文字列に対して部分文字列が最後に現れる位置を効率的に一括取得できます。なお、類似メソッドのnp.char.index()は左側から検索して最小のインデックスを返す点に注意してください。また、部分文字列が存在しない場合はValueErrorが発生するため、実務ではtry-exceptによる例外処理と組み合わせて使用することをおすすめします。