运行Python代码报'generator object is not subscriptable'错误求助

• 该错误的核心触发原因是调用get_unique_value函数时，实际传入的state参数并非预期的3x3 numpy数组，而是生成器对象。生成器对象未实现下标访问逻辑，无法使用[i]/[i][j]语法读取元素，因此执行到state[i][j]时直接抛出类型错误。
• 常见的误操作场景是传参前将数组用圆括号推导式包装，比如state = (x for x in 原始数组)，这种写法生成的就是生成器对象，而非数组。

• 首先排查传参逻辑，确保传入的state是合法numpy数组，可在函数入口增加校验代码快速定位问题：

import numpy as np
def get_unique_value(state):
    # 调试用校验，确认问题后可删除
    assert isinstance(state, np.ndarray), f"state类型异常：当前为{type(state)}，预期为numpy数组"
    assert state.shape == (3,3), f"state维度异常：当前为{state.shape}，预期为(3,3)"
    
    unique_value = 0
    for i in range(3):
        for j in range(3):
            # numpy数组推荐用[i,j]的索引方式，效率更高
            unique_value += int((10 ** (i*3 + j)) * state[i,j])
    return unique_value

• 如果确认是误将生成器传入，提前将生成器转为numpy数组即可，假设生成器变量为state_gen，转数组代码为state = np.array(list(state_gen))。
• 可选优化：用numpy向量化操作替代双层循环，代码更简洁执行效率更高：

def get_unique_value(state):
    power_mat = 10 ** np.arange(9).reshape(3,3)
    return int((state * power_mat).sum())

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