遍历类对象列表，返回该对象不可迭代？

Question

我有一个初始课程：

class foo:
    def __init__(self, a, b):
        self.a = a
        self.b = b

还有另一个使用 foo 类的类：

class bar:
    def __init__(self, foos):
        self.foos = sorted(foos, key=attrgetter('a'))

其中foos 是foo 的列表。我现在想要列出foo 的列表，看起来像：

lofoos = [[foo1, foo2, foo3], [foo4, foo5, foo6] ...]

我想使用地图功能来做到这一点：

list(map(lambda foos: bar(foos), lofoos))

但这会返回错误：

TypeError: iter() returned non-iterator of type 'foo'.  

有没有简单的解决方案？

Answer 1

问题是你得到了bar 个人foo 而不是foos 的列表，一个放置良好的打印揭示了问题

from operator import attrgetter

class foo:
   def __init__(self, a, b):
      self.a = a
      self.b = b
   def __repr__(self):
      return "{0.__class__.__name__}({0.a},{0.b})".format(self)

class bar:
   def __init__(self, foos):
      print("foos=",foos)
      self.foos = sorted(foos, key=attrgetter('a'))
   def __repr__(self):
      return "{0.__class__.__name__}({0.foos})".format(self)

lofoos = [[foo(1,0), foo(2,0), foo(3,0)], [foo(4,1), foo(5,1), foo(6,1)]]
print("test list of lists of foo")
print(list(map(lambda foos: bar(foos), lofoos)))
print("\n")
print("test list of foo")
print(list(map(lambda foos: bar(foos), lofoos[0])))

输出

test list of lists of foo
foos= [foo(1,0), foo(2,0), foo(3,0)]
foos= [foo(4,1), foo(5,1), foo(6,1)]
[bar([foo(1,0), foo(2,0), foo(3,0)]), bar([foo(4,1), foo(5,1), foo(6,1)])]


test list of foo
foos= foo(1,0)
Traceback (most recent call last):
  File "C:\Users\David\Documents\Python Scripts\stackoverflow_test.py", line 24, in <module>
    print(list(map(lambda foos: bar(foos), lofoos[0])))
  File "C:\Users\David\Documents\Python Scripts\stackoverflow_test.py", line 24, in <lambda>
    print(list(map(lambda foos: bar(foos), lofoos[0])))
  File "C:\Users\David\Documents\Python Scripts\stackoverflow_test.py", line 15, in __init__
    self.foos = sorted(foos, key=attrgetter('a'))
TypeError: 'foo' object is not iterable
>>> 

记住map(fun,[a,b,c]) 所做的是产生[fun(a),fun(b),fun(c)]

因此，在您的代码中的某个地方，您最终会在 foo 列表中进行映射，而不是在 foo 列表中进行映射