使用for循环简化数据(Python)

Question

我试图简化代码:

            header = []
            header.append(header1)
            header.append(header2)                
            header.append(header3)
            header.append(header4)
            header.append(header5)
            header.append(header6)

哪里:

            header1 = str(input.headerOut1)
            header2 = str(input.headerOut2)
            header3 = str(input.headerOut3)
            header4 = str(input.headerOut4)
            header5 = str(input.headerOut5)
            header6 = str(input.headerOut6)

我本来想使用for循环,比如:

   headerList = []
   for i in range(6)
          headerList.append(header+i) 

但是,python不会识别标头+ i代表字符串header1.有没有办法简化这个代码或让for循环工作？非常感谢!

Answer 1

您应该将数据构建为列表或字典,如下所示:

input.headerOut[1]
input.headerOut[2]
# etc.

这将使这更容易,更Pythonic.但你可以做你想要的事情getattr:

headerList = []
for i in range(1, 7):
    header = str(getattr(input, 'headerOut%d' % i))
    headerList.append(header)

Answer 2

header = [str(getattr(input, "headerOut%d" % x)) for x in range(1,7)]