Python中使用NumPy数组触发IndexError的问题排查求助

我编写了一段Python代码，在使用NumPy数组时持续触发IndexError，但无法定位问题原因。代码如下：

import numpy as np

mainMenu = np.array(['Load new data', 'Check for data errors', 'Generate plots', 'Display list of grades', 'Quit'])
if choice == 2 or choice == 3 or choice == 4:
    print("Please load data before you {:s}".format(mainMenu[choice-1]))
    continue

执行上述代码的后半部分时出现错误，报错信息为：

IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices

请问有人能找出问题所在吗？

解答 #

这个问题的核心原因大概率是**choice变量不是整数类型**！

NumPy数组对索引的类型要求比普通Python列表严格得多——它只接受整数、切片、布尔数组等类型的索引。如果choice是字符串（比如用户直接输入的字符串形式数字）或者浮点数（比如2.0这种看起来像整数的浮点数），都会触发这个错误。举个例子：

• 如果choice是字符串"2"，执行choice-1时会直接抛出类型错误；
• 如果choice是浮点数2.0，Python列表允许[1,2,3][2.0]这种写法，但NumPy数组会直接拒绝这种非整数类型的索引。

解决方法：

1. 检查choice的类型：在判断语句前加一行print(type(choice))，确认它是不是int类型。
2. 确保输入转换为整数：如果choice来自用户输入，一定要把输入转换成整数，比如：

   choice = int(input("Enter your menu choice: "))
   

3. 增加类型校验：在使用索引前先验证choice的类型，避免意外情况：

   if isinstance(choice, int) and choice in (2, 3, 4):
       print("Please load data before you {:s}".format(mainMenu[choice-1]))
       continue
   

这样就能彻底解决这个索引错误啦！

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