无法通过数值匹配从Pandas Series获取索引的问题排查

问题场景 #

基于数字与字符列表创建了Pandas Series对象charSeries，内容如下：

0     0
1     1
2     2
3     3
4     4
5     5
6     6
7     7
8     8
9     9
10    B
11    C
12    D
13    F
14    G
15    H
16    J
17    K
18    L
19    M
20    N
21    P
22    Q
23    R
24    S
25    T
26    V
27    W
28    X
29    Y
30    Z

生成随机字符数字混合五位数串code_core = "Q5YUO"，遍历其中每个字符，用下面的代码获取对应索引：

for character in code_core:
    index_tosum = charSeries[charSeries == character].index(0)

字母字符能正常匹配，但碰到数字字符（比如'5'）就触发索引越界错误，报错信息如下：

return getitem(key)
IndexError: index 0 is out of bounds for axis 0 with size 0

明明charSeries里有对应的数字，为啥匹配结果为空？

问题根源 #

类型不匹配：charSeries里的数字是整数类型（比如5是int），但code_core里的数字是字符串类型（比如'5'是str）。用字符串'5'去匹配整数5，当然找不到对应值，筛选结果为空后调用.index(0)自然就报错了。另外还有个语法小问题：.index是Pandas的Index对象，取元素要用方括号.index[0]，不是圆括号.index(0)。

解决方案 #

方法1：统一元素为字符串类型 #

创建charSeries时，把所有元素都转成字符串：

# 假设原列表包含数字和字母
nums_and_chars = [str(i) for i in range(10)] + ['B','C','D','F','G','H','J','K','L','M','N','P','Q','R','S','T','V','W','X','Y','Z']
charSeries = pd.Series(nums_and_chars)

这样后续用字符串匹配就能直接找到对应索引。

方法2：遍历动态转换字符类型 #

遍历code_core时，判断字符是否为数字，是则转成整数再匹配，字母保持字符串：

for character in code_core:
    target = int(character) if character.isdigit() else character
    index_tosum = charSeries[charSeries == target].index[0]

效率优化建议 #

如果需要频繁通过值查索引，建议提前做个值到索引的映射字典，比每次筛选Series快得多：

# 构建映射字典
value_to_index = charSeries.reset_index().set_index(0)['index'].to_dict()

# 后续直接通过字典取值
for character in code_core:
    target = int(character) if character.isdigit() else character
    index_tosum = value_to_index[target]

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