如何获取PandasArray对象中指定元素的索引？

针对你创建的PandasArray对象a = pd.array(['a', 'b'], dtype=str)，你可以通过以下几种方式获取元素'a'的索引：

方法1：使用numpy的where函数 #

PandasArray兼容numpy数组操作，直接用布尔匹配结合np.where即可得到所有匹配元素的索引：

import pandas as pd
import numpy as np

a = pd.array(['a', 'b'], dtype=str)
# 获取所有匹配'a'的索引，返回numpy数组
match_indices = np.where(a == 'a')[0]
# 若只需要第一个匹配的索引，取数组第一个元素
first_index = match_indices[0]

方法2：转换为Series后操作 #

将PandasArray转为Series，利用Pandas的索引功能查找：

import pandas as pd

a = pd.array(['a', 'b'], dtype=str)
s = pd.Series(a)
# 通过布尔筛选获取索引对象
match_indices = s[s == 'a'].index
# 取第一个匹配索引
first_index = match_indices[0]

如果数组中存在多个'a'元素，上述方法会返回所有对应的索引，你可以根据需求遍历或提取特定位置的索引。

内容的提问来源于stack exchange，提问作者Tiancheng Liu