pythonのSeriesの要素を検索し、対応するインデックスを取得

Question

PandasのSeriesの特定の要素を検索して
対応するインデックスを取り出す方法を教えてください

0　Japan
1　America
2　China
3　Canada

例えばAmericaを指定したら1が返ってくるようにしたいです

Answer 1

どちらが良い悪いは得にありませんが、2案提示します。
両案とも対応する要素が1件以上ある場合は最初のindex、ない場合はNoneを返します。

import pandas as pd

record = pd.Series(['Japan', 'America', 'China', 'Canada'])
print(record)
# 0      Japan
# 1    America
# 2      China
# 3     Canada
# dtype: object

# 案1
def find_index1(item):
    return record.where(record == item).first_valid_index()

print(find_index1('America')) # -> 1
print(find_index1('Canada')) # -> 3
print(find_index1('France')) # -> None

# 案2
def find_index2(item):
    match_items = record[record == item]
    if match_items.empty:
        return None
    return match_items.index[0]

print(find_index2('America')) # -> 1
print(find_index2('Canada')) # -> 3
print(find_index2('France')) # None

Comments

[kenoh]

上記方法で解決しました。ありがとうございました。