如何通过Python特殊方法让ctypes结构体的for迭代在指定长度停止

要让你的MyCClass实例支持按n_values指定的长度迭代，只需要给类添加一个__iter__方法，用生成器来控制迭代范围即可。

默认情况下，Python对序列类型对象迭代时，如果没有自定义__iter__，会依赖__getitem__从索引0开始逐个获取元素，直到抛出IndexError。但你的values是固定长度为5的数组，所以会遍历完所有5个元素，不会理会你定义的__len__返回值。

修改后的完整代码如下：

import ctypes

class MyCClass(ctypes.Structure):
    _fields_ = [('n_values', ctypes.c_int),
                ('values', ctypes.c_double * 5)]
    
    def __repr__(self):
        return """n_values : {0}, values : {1}""".format(self.n_values,
                                                         self.values)
    
    def __len__(self):
        return self.n_values
    
    def __getitem__(self, key):
        # 新增越界检查，和普通序列行为对齐
        if key >= self.n_values:
            raise IndexError("Index out of bounds")
        return self.values[key]
    
    def __iter__(self):
        # 生成器只迭代有效范围内的元素
        for i in range(self.n_values):
            yield self.values[i]

测试验证 #

# 创建实例并设置有效数据
obj = MyCClass()
obj.n_values = 3
obj.values[0] = 0.0
obj.values[1] = 1.0
obj.values[2] = 2.0

# 现在可以直接按有效长度迭代
for value in obj:
    print(value)
# 输出：
# 0.0
# 1.0
# 2.0

我还在__getitem__里补充了越界检查，这样当你尝试访问超过n_values的索引时，会抛出标准的IndexError，避免返回无效的0值，让类的行为更贴近Python原生序列。

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