PythonでPandas Seriesに含まれる最長の文字列を検索する方法

はじめに

この記事では、PythonとPandasを使って、Series（シリーズ）内の文字列の中から最も長い文字列を見つけるプログラムを作成する方法を解説します。

入力データ

以下のようなSeriesが与えられたと仮定します。

["one", "two", "eleven", "pomegranates", "three"]

この中で最も長い文字列は "pomegranates" です。これを出力として取得できるプログラムを目指します。

解決のアプローチ

この問題は、以下の手順で解決できます。

• PandasのSeriesを定義します。
• 最大長を記録する変数「maxlen」の初期値を0に設定します。
• 最長の文字列を格納する変数「maxstr」の初期値を空文字列に設定します。
• forループを使ってSeries内の値を1つずつ取り出し、if条件で文字列の長さを比較します。

ループ処理のコードは次のようになります。

for i in res:
    if(len(i) > maxlen):
        maxlen = len(i)
        maxstr = i

• 最後に、「maxstr」変数に格納された値を出力します。

実装例

実際のコードを見て、理解を深めましょう。

import pandas as pd
res = pd.Series(["one","two","eleven","pomegranates","three"])
maxlen = len(res[0])
maxstr = ""
for i in res:
    if(len(i) > maxlen):
        maxlen = len(i)
        maxstr = i
print(maxstr)

出力結果

pomegranates

コードのポイント

このプログラムでは、初期値として最初の要素の長さを「maxlen」に設定しています。その後、各要素の長さが現在の最大長を上回るたびに、「maxlen」と「maxstr」を更新していきます。すべての要素を走査し終えた時点で、「maxstr」にはSeries内で最も長い文字列が格納されているという仕組みです。

なお、より簡潔に書きたい場合は、組み込み関数 max() を使って max(res, key=len) とすることでも同じ結果が得られます。状況に応じて使い分けると良いでしょう。