如何处理带有嵌套列表和字符串元素的列表？

Question

我想展平一个包含嵌套元素而不是嵌套元素的列表。从this 解决方案我已经尝试过，如果mylist 中的所有元素都是列表，它就可以工作，但是 在mylist 我有简单的文本字符串和嵌套列表。

我的清单是这样的：

mylist = [
        'tz',
        '7',
        ['a', 'b', 'c'], 
        [['2'], ['4', 'r'], ['34']], 
        [['7'], ['3',  ['2', ['1']]], ['9']], 
        [['11',['7','w']], 'U1', ['0']]
    ]

我当前的代码是这样的，得到以下错误：

import collections#.abc

def flatten(l):
    for el in l:
        if isinstance(el, collections.Iterable) and not isinstance(el, (str, bytes)):
            yield from flatten(el)
        else:
            yield el

            
mylist1=[list(flatten(sublist)) if type(sublist) is list else sublist for sublist in mylist]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <listcomp>
  File "<stdin>", line 3, in flatten
TypeError: isinstance() arg 2 must be a type or tuple of types
>>>

我的预期输出是这样的

mylist1 = [
        'tz',
        '7',
        ['a', 'b', 'c'], 
        ['2', '4', 'r','34'], 
        ['7','3','2','1','9'],
        ['11','7','w','U1','0']
    ]

缺少什么来解决这个问题？谢谢。

更新

现在，@Alok 建议的代码出现此错误

>>> for item in mylist:
...     # if the item in mylist is a list, pass it to your flatten method else
...     # add it to the final list
...     if isinstance(item, list):
...         final_list.append(list(flatten(item)))
...     else:
...         final_list.append(item)
...
Traceback (most recent call last):
  File "<stdin>", line 5, in <module>
  File "<stdin>", line 3, in flatten
TypeError: isinstance() arg 2 must be a type or tuple of types

Answer 1

问题在于我将指出的一些实例：

1. Python 3.x 不再关注collections.Iterable，您需要从collections.abc 导入项目，并始终为任何导入错误执行try-catch

try:
    from collections.abc import Iterable
except ImportError:
    from collections import Iterable 

1. 未正确使用您的 flatten 方法，该方法返回数据，但您必须将其以列表的形式存储并将其附加到最终答案列表 您的子数组只能传递到此方法中

data.append(list(flatten(item)))

最终解决方案：

try:
    from collections.abc import Iterable
except ImportError:
    from collections import Iterable 

mylist = [
    'tz',
    '7',
    ['a', 'b', 'c'], 
    [['2'], ['4', 'r'], ['34']], 
    [['7'], ['3',  ['2', ['1']]], ['9']], 
    [['11',['7','w']], 'U1', ['0']]
]

def flatten(l):
    for el in l:
        if isinstance(el, Iterable) and not isinstance(el, (str, bytes)):
            yield from flatten(el)
        else:
            yield el
            
final_list = []
for item in mylist:
    # if the item in mylist is a list, pass it to your flatten method else
    # add it to the final list
    if isinstance(item, type([])): final_list.append(list(flatten(item)))
    else: final_list.append(item)
    
print(final_list)

输出

['tz', '7', ['a', 'b', 'c'], ['2', '4', 'r', '34'], ['7', '3', '2', '1', '9'], ['11', '7', 'w', 'U1', '0']]

希望你可以通过这种方式实现你想要的输出。