Python中通过for循环遍历对象属性提取数据的技术问询

你这段代码的核心问题在于动态属性访问的方式错误——你用了.来访问属性，但attribute是循环中的变量，Python会把它当成对象的固定属性名（也就是去找名为attribute的属性），而不是你循环列表里的attr1、attr2这些目标属性。

修正方案：使用getattr()函数动态获取属性 #

Python提供了getattr()内置函数，专门用来通过字符串动态访问对象的属性。把你的代码改成下面这样就能正常工作了：

def calculate_something(dictionaryWithObjects):
    for attribute in ['attr1', 'attr2', 'attr3']:
        # 用getattr动态获取指定属性，参数为对象和属性名字符串
        dataA1, dataA2 = getattr(dictionaryWithObjects['Object1'], attribute)
        dataB1, dataB2 = getattr(dictionaryWithObjects['Object2'], attribute)
        dataC1, dataC2 = getattr(dictionaryWithObjects['Object3'], attribute)
        
        # 在这里添加你的计算逻辑，比如基于提取的数据进行运算
        # 示例：计算三个对象对应属性第一个值的总和
        sum_first_values = dataA1 + dataB1 + dataC1
        print(f"Sum of first values for {attribute}: {sum_first_values}")

额外优化建议 #

• 处理属性不存在的情况：如果不确定所有对象都有目标属性，可以给getattr()加第三个参数作为默认值，避免报错：

  # 假设默认返回(0, 0)当属性不存在时
  dataA1, dataA2 = getattr(dictionaryWithObjects['Object1'], attribute, (0, 0))
  

• 简化重复代码：如果需要处理的对象很多，可以把对象列表提出来循环，减少重复代码：

  def calculate_something(dictionaryWithObjects):
      objects = [dictionaryWithObjects['Object1'], dictionaryWithObjects['Object2'], dictionaryWithObjects['Object3']]
      for attribute in ['attr1', 'attr2', 'attr3']:
          all_data = [getattr(obj, attribute) for obj in objects]
          # 然后从all_data里提取各个dataX1、dataX2
          dataA1, dataA2 = all_data[0]
          dataB1, dataB2 = all_data[1]
          dataC1, dataC2 = all_data[2]
          # 后续计算逻辑...
  

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