Python-迭代任何类型的列表结构

Question

对不起，如果已经回答了，我对python非常新，我环顾四周，在这里找到了此页，这对我有所帮助，但我'm仍然卡住。

我正在尝试使任何类型的输入在脚本中工作，我让它适用于单个项目和一个项目列表，但是我发现很难让它适用于列表清单。

我已经根据注释编辑了代码，以使其更有意义：

Input = [[1,2,3],[4,5,6],[7,8,9]]
if isinstance(Input, list):
    Input = Input
else:
    Input = [Input]
listout = []

for x in Input:
    listout.append(x+2)
print (listout)

返回：第12行，在 listout.append（x 2）typeError：只能连接列表（不是" int"）

例如，如果输入= 1或输入= [1,2,3,4]，但对于上述。

我希望输出看起来像以下列表的列表：

[[3,4,5],[6,7,8],[9,10,11]]

我试图首先从嵌套列表中列出一个平面列表，但我想保留输出的列表结构。

感谢大家的阅读，

TJ

Answer 1

您可以考虑numpy：

>>> import numpy as np
>>> a=np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> a
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
>>> a+=2
>>> a
array([[ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11]])
>>> a.tolist()
[[3, 4, 5], [6, 7, 8], [9, 10, 11]]

---

如果您不能使用numpy，则需要编写一个任意嵌套的递归程序：

def nested_sum(e,n):
    if isinstance(e, list):
        return [nested_sum(x, n) for x in e]
    else:
        return e+n    
>>> nested_sum([1,[2,3],[4,5,6],[7,[8],9]], 2)
[3, [4, 5], [6, 7, 8], [9, [10], 11]]

如果您只有两个层次的嵌套（如示例中所示），则可以做一个列表理解：

>>> li=[[1,2,3],[4,5,6],[7,8,9]]
>>> [[e+2 for e in sl] for sl in li]
[[3, 4, 5], [6, 7, 8], [9, 10, 11]]