ฉันสงสัยว่ามีทางลัดในการสร้างรายการอย่างง่ายจากรายการใน Python หรือไม่

ฉันสามารถทำได้ในforวง แต่อาจจะมี "หนึ่งซับ" ที่เจ๋ง ๆ ? ฉันลองด้วยreduce()แต่ฉันได้รับข้อผิดพลาด

รหัส

l = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
reduce(lambda x, y: x.extend(y), l)

ข้อความผิดพลาด

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <lambda>
AttributeError: 'NoneType' object has no attribute 'extend'

ได้รับรายชื่อของรายการl,

flat_list = [item for sublist in l for item in sublist]

ซึ่งหมายความว่า:

flat_list = []
for sublist in l:
    for item in sublist:
        flat_list.append(item)

เร็วกว่าทางลัดที่โพสต์ไปแล้ว ( lคือรายการที่จะเรียบ)

นี่คือฟังก์ชั่นที่เกี่ยวข้อง:

flatten = lambda l: [item for sublist in l for item in sublist]

ตามหลักฐานคุณสามารถใช้timeitโมดูลในไลบรารีมาตรฐาน:

$ python -mtimeit -s'l=[[1,2,3],[4,5,6], [7], [8,9]]*99' '[item for sublist in l for item in sublist]'
10000 loops, best of 3: 143 usec per loop
$ python -mtimeit -s'l=[[1,2,3],[4,5,6], [7], [8,9]]*99' 'sum(l, [])'
1000 loops, best of 3: 969 usec per loop
$ python -mtimeit -s'l=[[1,2,3],[4,5,6], [7], [8,9]]*99' 'reduce(lambda x,y: x+y,l)'
1000 loops, best of 3: 1.1 msec per loop

คำอธิบาย: ทางลัดที่อิงตาม+(รวมถึงการใช้โดยนัยsum) มีความจำเป็นO(L**2)เมื่อมีรายการย่อย L - เนื่องจากรายการผลกลางจะได้รับอีกต่อไปในแต่ละขั้นตอนจะมีการจัดสรรวัตถุรายการผลลัพธ์ระดับกลางใหม่และรายการทั้งหมด ในผลลัพธ์ระดับกลางก่อนหน้าจะต้องคัดลอก (เช่นเดียวกับรายการใหม่ที่เพิ่มเข้ามาในตอนท้าย) ดังนั้นเพื่อความง่ายและโดยไม่สูญเสียความเป็นปรกติจริง ๆ สมมติว่าคุณมีรายการย่อย L ของแต่ละรายการ: รายการไอเท็มแรกของฉันจะถูกคัดลอกกลับไปกลับมา L-1 คูณสองรายการไอเท็ม L-2 ครั้งและอื่น ๆ ; จำนวนสำเนาคือผมครั้งผลรวมของ x สำหรับ x จาก 1 ถึง L I * (L**2)/2ยกเว้นคือ,

ความเข้าใจในรายการเพียงแค่สร้างหนึ่งรายการครั้งเดียวและคัดลอกแต่ละรายการผ่าน (จากที่อยู่เดิมไปยังรายการผลลัพธ์) เช่นกัน

คุณสามารถใช้itertools.chain():

import itertools
list2d = [[1,2,3], [4,5,6], [7], [8,9]]
merged = list(itertools.chain(*list2d))

หรือคุณสามารถใช้itertools.chain.from_iterable()ซึ่งไม่ต้องการเปิดรายการด้วยตัว*ดำเนินการ :

import itertools
list2d = [[1,2,3], [4,5,6], [7], [8,9]]
merged = list(itertools.chain.from_iterable(list2d))

หมายเหตุจากผู้เขียน : นี่ไม่มีประสิทธิภาพ แต่สนุกเพราะmonoidsน่ากลัว มันไม่เหมาะสำหรับการผลิตรหัส Python

>>> sum(l, [])
[1, 2, 3, 4, 5, 6, 7, 8, 9]

นี่เป็นผลรวมขององค์ประกอบที่ผ่าน iterable ในอาร์กิวเมนต์แรกซึ่งถือว่าอาร์กิวเมนต์ที่สองเป็นค่าเริ่มต้นของผลรวม (หากไม่ได้รับ0จะถูกใช้แทนและกรณีนี้จะทำให้คุณมีข้อผิดพลาด)

เพราะคุณจะได้ข้อสรุปรายการที่ซ้อนกันคุณได้รับจริง[1,3]+[2,4]เป็นผลจากการที่จะมีค่าเท่ากับsum([[1,3],[2,4]],[])[1,3,2,4]

โปรดทราบว่าใช้งานได้กับรายการของรายการเท่านั้น สำหรับรายการของรายการคุณจะต้องมีวิธีแก้ไขปัญหาอื่น

ฉันทดสอบวิธีแก้ปัญหาที่แนะนำส่วนใหญ่ด้วยperfplot (โครงการสัตว์เลี้ยงของฉันเป็นหลักห่อหุ้มรอบ ๆtimeit) และพบว่า

functools.reduce(operator.iconcat, a, [])

เพื่อเป็นทางออกที่เร็วที่สุดทั้งเมื่อมีรายการขนาดเล็กจำนวนมากและรายการยาวไม่กี่รายการที่ต่อกัน ( operator.iaddเร็วพอ ๆ กัน)

รหัสในการทำซ้ำพล็อต:

import functools
import itertools
import numpy
import operator
import perfplot


def forfor(a):
    return [item for sublist in a for item in sublist]


def sum_brackets(a):
    return sum(a, [])


def functools_reduce(a):
    return functools.reduce(operator.concat, a)


def functools_reduce_iconcat(a):
    return functools.reduce(operator.iconcat, a, [])


def itertools_chain(a):
    return list(itertools.chain.from_iterable(a))


def numpy_flat(a):
    return list(numpy.array(a).flat)


def numpy_concatenate(a):
    return list(numpy.concatenate(a))


perfplot.show(
    setup=lambda n: [list(range(10))] * n,
    # setup=lambda n: [list(range(n))] * 10,
    kernels=[
        forfor,
        sum_brackets,
        functools_reduce,
        functools_reduce_iconcat,
        itertools_chain,
        numpy_flat,
        numpy_concatenate,
    ],
    n_range=[2 ** k for k in range(16)],
    xlabel="num lists (of length 10)",
    # xlabel="len lists (10 lists total)"
)

from functools import reduce #python 3

>>> l = [[1,2,3],[4,5,6], [7], [8,9]]
>>> reduce(lambda x,y: x+y,l)
[1, 2, 3, 4, 5, 6, 7, 8, 9]

extend()วิธีการในตัวอย่างของคุณปรับเปลี่ยนxแทนที่จะกลับค่าที่มีประโยชน์ (ซึ่งreduce()คาดว่า)

วิธีที่เร็วกว่าในการทำreduceเวอร์ชันจะเป็น

>>> import operator
>>> l = [[1,2,3],[4,5,6], [7], [8,9]]
>>> reduce(operator.concat, l)
[1, 2, 3, 4, 5, 6, 7, 8, 9]

อย่าบูรณาการล้อถ้าคุณใช้Django :

>>> from django.contrib.admin.utils import flatten
>>> l = [[1,2,3], [4,5], [6]]
>>> flatten(l)
>>> [1, 2, 3, 4, 5, 6]

... นุ่น :

>>> from pandas.core.common import flatten
>>> list(flatten(l))

... Itertools :

>>> import itertools
>>> flatten = itertools.chain.from_iterable
>>> list(flatten(l))

... Matplotlib

>>> from matplotlib.cbook import flatten
>>> list(flatten(l))

... Unipath :

>>> from unipath.path import flatten
>>> list(flatten(l))

... Setuptools :

>>> from setuptools.namespaces import flatten
>>> list(flatten(l))

นี่คือวิธีการทั่วไปที่ใช้กับตัวเลข , สตริง , ซ้อนกันรายการและผสมภาชนะบรรจุ

รหัส

#from typing import Iterable 
from collections import Iterable                            # < py38


def flatten(items):
    """Yield items from any nested iterable; see Reference."""
    for x in items:
        if isinstance(x, Iterable) and not isinstance(x, (str, bytes)):
            for sub_x in flatten(x):
                yield sub_x
        else:
            yield x

หมายเหตุ :

• ใน Python 3 yield from flatten(x)สามารถแทนที่for sub_x in flatten(x): yield sub_x
• ใน Python 3.8 คลาสพื้นฐานที่เป็นนามธรรมจะถูกย้ายจากcollection.abcไปยังtypingโมดูล

การสาธิต

lst = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
list(flatten(lst))                                         # nested lists
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

mixed = [[1, [2]], (3, 4, {5, 6}, 7), 8, "9"]              # numbers, strs, nested & mixed
list(flatten(mixed))
# [1, 2, 3, 4, 5, 6, 7, 8, '9']

การอ้างอิง

• วิธีการแก้ปัญหานี้ได้รับการแก้ไขจากสูตรในBeazley, D. และ B. Jones สูตร 4.14, Python Cookbook เล่มที่ 3, O'Reilly Media Inc. Sebastopol, CA: 2013
• พบโพสต์ SOก่อนหน้าอาจเป็นการสาธิตดั้งเดิม

หากคุณต้องการเรียบโครงสร้างข้อมูลที่คุณไม่ทราบว่ามันซ้อนกันลึกเพียงใดคุณสามารถใช้¹iteration_utilities.deepflatten

>>> from iteration_utilities import deepflatten

>>> l = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
>>> list(deepflatten(l, depth=1))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

>>> l = [[1, 2, 3], [4, [5, 6]], 7, [8, 9]]
>>> list(deepflatten(l))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

มันเป็นตัวกำเนิดดังนั้นคุณต้องส่งผลลัพธ์ไปที่listหรือย้ำอย่างชัดเจน

หากต้องการแบนเพียงระดับเดียวและหากรายการแต่ละรายการเป็นตัวกำหนดเองคุณสามารถใช้iteration_utilities.flattenสิ่งที่เป็นเพียงเสื้อคลุมบาง ๆitertools.chain.from_iterable:

>>> from iteration_utilities import flatten
>>> l = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
>>> list(flatten(l))
[1, 2, 3, 4, 5, 6, 7, 8, 9]

เพียงเพื่อเพิ่มการกำหนดเวลา (ตามคำตอบของ Nico Schlömerที่ไม่รวมฟังก์ชั่นที่แสดงในคำตอบนี้):

มันเป็นพล็อตการบันทึกการทำงานเพื่อรองรับค่าที่หลากหลาย สำหรับเหตุผลเชิงคุณภาพ: ต่ำกว่าดีกว่า

ผลการศึกษาพบว่าหากมี iterable iterables ภายในเพียงไม่กี่แล้วsumจะเร็วที่สุด แต่สำหรับ iterables ยาวเพียงitertools.chain.from_iterable, iteration_utilities.deepflattenหรือความเข้าใจที่ซ้อนกันมีประสิทธิภาพการทำงานที่เหมาะสมกับitertools.chain.from_iterableการเป็นที่เร็วที่สุด (เท่าที่สังเกตเห็นแล้วโดยนิโก้Schlömer)

from itertools import chain
from functools import reduce
from collections import Iterable  # or from collections.abc import Iterable
import operator
from iteration_utilities import deepflatten

def nested_list_comprehension(lsts):
    return [item for sublist in lsts for item in sublist]

def itertools_chain_from_iterable(lsts):
    return list(chain.from_iterable(lsts))

def pythons_sum(lsts):
    return sum(lsts, [])

def reduce_add(lsts):
    return reduce(lambda x, y: x + y, lsts)

def pylangs_flatten(lsts):
    return list(flatten(lsts))

def flatten(items):
    """Yield items from any nested iterable; see REF."""
    for x in items:
        if isinstance(x, Iterable) and not isinstance(x, (str, bytes)):
            yield from flatten(x)
        else:
            yield x

def reduce_concat(lsts):
    return reduce(operator.concat, lsts)

def iteration_utilities_deepflatten(lsts):
    return list(deepflatten(lsts, depth=1))


from simple_benchmark import benchmark

b = benchmark(
    [nested_list_comprehension, itertools_chain_from_iterable, pythons_sum, reduce_add,
     pylangs_flatten, reduce_concat, iteration_utilities_deepflatten],
    arguments={2**i: [[0]*5]*(2**i) for i in range(1, 13)},
    argument_name='number of inner lists'
)

b.plot()

^{1 ข้อสงวนสิทธิ์: ฉันเป็นผู้แต่งห้องสมุดดังกล่าว}

ฉันเอาคำพูดของฉันกลับมา ผลรวมไม่ใช่ผู้ชนะ แม้ว่าจะเร็วกว่าเมื่อรายการมีขนาดเล็ก แต่ประสิทธิภาพลดลงอย่างมีนัยสำคัญกับรายการขนาดใหญ่

>>> timeit.Timer(
        '[item for sublist in l for item in sublist]',
        'l=[[1, 2, 3], [4, 5, 6, 7, 8], [1, 2, 3, 4, 5, 6, 7]] * 10000'
    ).timeit(100)
2.0440959930419922

เวอร์ชั่นรวมยังคงใช้งานได้นานกว่าหนึ่งนาทีและยังไม่ได้ประมวลผล!

สำหรับรายการสื่อ:

>>> timeit.Timer(
        '[item for sublist in l for item in sublist]',
        'l=[[1, 2, 3], [4, 5, 6, 7, 8], [1, 2, 3, 4, 5, 6, 7]] * 10'
    ).timeit()
20.126545906066895
>>> timeit.Timer(
        'reduce(lambda x,y: x+y,l)',
        'l=[[1, 2, 3], [4, 5, 6, 7, 8], [1, 2, 3, 4, 5, 6, 7]] * 10'
    ).timeit()
22.242258071899414
>>> timeit.Timer(
        'sum(l, [])',
        'l=[[1, 2, 3], [4, 5, 6, 7, 8], [1, 2, 3, 4, 5, 6, 7]] * 10'
    ).timeit()
16.449732065200806

ใช้รายการขนาดเล็กและ timeit: number = 1000000

>>> timeit.Timer(
        '[item for sublist in l for item in sublist]',
        'l=[[1, 2, 3], [4, 5, 6, 7, 8], [1, 2, 3, 4, 5, 6, 7]]'
    ).timeit()
2.4598159790039062
>>> timeit.Timer(
        'reduce(lambda x,y: x+y,l)',
        'l=[[1, 2, 3], [4, 5, 6, 7, 8], [1, 2, 3, 4, 5, 6, 7]]'
    ).timeit()
1.5289170742034912
>>> timeit.Timer(
        'sum(l, [])',
        'l=[[1, 2, 3], [4, 5, 6, 7, 8], [1, 2, 3, 4, 5, 6, 7]]'
    ).timeit()
1.0598428249359131

ดูเหมือนจะมีความสับสนด้วยoperator.add! เมื่อคุณเพิ่มสองรายการเข้าด้วยกันคำที่ถูกต้องคือconcatไม่เพิ่ม operator.concatคือสิ่งที่คุณต้องการใช้

หากคุณคิดว่าการใช้งานเป็นเรื่องง่ายเหมือนนี้ ::

>>> from functools import reduce
>>> list2d = ((1, 2, 3), (4, 5, 6), (7,), (8, 9))
>>> reduce(operator.concat, list2d)
(1, 2, 3, 4, 5, 6, 7, 8, 9)

คุณจะเห็นการลดลงของลำดับลำดับดังนั้นเมื่อคุณจัดหา tuple คุณจะได้รับ tuple กลับมา ลองทำรายการ ::

>>> list2d = [[1, 2, 3],[4, 5, 6], [7], [8, 9]]
>>> reduce(operator.concat, list2d)
[1, 2, 3, 4, 5, 6, 7, 8, 9]

อ๊ะคุณกลับรายการแล้ว

วิธีการเกี่ยวกับประสิทธิภาพ ::

>>> list2d = [[1, 2, 3],[4, 5, 6], [7], [8, 9]]
>>> %timeit list(itertools.chain.from_iterable(list2d))
1000000 loops, best of 3: 1.36 µs per loop

from_iterableค่อนข้างเร็ว! concatแต่มันก็เปรียบเทียบเพื่อลดไม่มี

>>> list2d = ((1, 2, 3),(4, 5, 6), (7,), (8, 9))
>>> %timeit reduce(operator.concat, list2d)
1000000 loops, best of 3: 492 ns per loop

เหตุใดคุณใช้ส่วนขยาย

reduce(lambda x, y: x+y, l)

ควรทำงานได้ดี

พิจารณาติดตั้งmore_itertoolsแพ็กเกจ

> pip install more_itertools

มันมาพร้อมกับการใช้งานสำหรับflatten( แหล่งที่มาจากสูตร itertools ):

import more_itertools


lst = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
list(more_itertools.flatten(lst))
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

ในเวอร์ชัน 2.4 คุณสามารถแบนตัวควบคุม iterables แบบซ้อนที่มีความซับซ้อนมากขึ้นmore_itertools.collapse( แหล่งที่มาสนับสนุนโดย abarnet)

lst = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
list(more_itertools.collapse(lst)) 
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

lst = [[1, 2, 3], [[4, 5, 6]], [[[7]]], 8, 9]              # complex nesting
list(more_itertools.collapse(lst))
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

เหตุผลที่ฟังก์ชั่นของคุณใช้งานไม่ได้เพราะส่วนขยายขยายอาร์เรย์เข้าที่และไม่ส่งคืน คุณยังสามารถคืนค่า x จากแลมบ์ดาโดยใช้บางสิ่งเช่นนี้:

reduce(lambda x,y: x.extend(y) or x, l)

หมายเหตุ: ส่วนขยายมีประสิทธิภาพมากกว่า + ในรายการ

def flatten(l, a):
    for i in l:
        if isinstance(i, list):
            flatten(i, a)
        else:
            a.append(i)
    return a

print(flatten([[[1, [1,1, [3, [4,5,]]]], 2, 3], [4, 5],6], []))

# [1, 1, 1, 3, 4, 5, 2, 3, 4, 5, 6]

เวอร์ชันเรียกซ้ำ

x = [1,2,[3,4],[5,[6,[7]]],8,9,[10]]

def flatten_list(k):
    result = list()
    for i in k:
        if isinstance(i,list):

            #The isinstance() function checks if the object (first argument) is an 
            #instance or subclass of classinfo class (second argument)

            result.extend(flatten_list(i)) #Recursive call
        else:
            result.append(i)
    return result

flatten_list(x)
#result = [1,2,3,4,5,6,7,8,9,10]

matplotlib.cbook.flatten() จะใช้งานได้กับรายการที่ซ้อนกันแม้ว่าจะซ้อนอยู่ลึกกว่าตัวอย่าง

import matplotlib
l = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
print(list(matplotlib.cbook.flatten(l)))
l2 = [[1, 2, 3], [4, 5, 6], [7], [8, [9, 10, [11, 12, [13]]]]]
print list(matplotlib.cbook.flatten(l2))

ผลลัพธ์:

[1, 2, 3, 4, 5, 6, 7, 8, 9]
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]

นี่คือ 18x เร็วกว่าขีดล่าง. _. แผ่:

Average time over 1000 trials of matplotlib.cbook.flatten: 2.55e-05 sec
Average time over 1000 trials of underscore._.flatten: 4.63e-04 sec
(time for underscore._)/(time for matplotlib.cbook) = 18.1233394636

คำตอบที่ยอมรับนั้นใช้ไม่ได้กับฉันเมื่อจัดการกับรายการข้อความที่มีความยาวผันแปรได้ นี่คือวิธีการสำรองที่ใช้งานได้สำหรับฉัน

l = ['aaa', 'bb', 'cccccc', ['xx', 'yyyyyyy']]

คำตอบที่ยอมรับซึ่งไม่ได้ผล:

flat_list = [item for sublist in l for item in sublist]
print(flat_list)
['a', 'a', 'a', 'b', 'b', 'c', 'c', 'c', 'c', 'c', 'c', 'xx', 'yyyyyyy']

โซลูชั่นใหม่ที่นำเสนอไม่ทำงานสำหรับฉัน:

flat_list = []
_ = [flat_list.extend(item) if isinstance(item, list) else flat_list.append(item) for item in l if item]
print(flat_list)
['aaa', 'bb', 'cccccc', 'xx', 'yyyyyyy']

คุณลักษณะที่ไม่ดีของฟังก์ชั่นของ Anil ด้านบนคือต้องการให้ผู้ใช้ระบุอาร์กิวเมนต์ที่สองด้วยตนเองเป็นรายการว่าง[]เสมอ นี่ควรเป็นค่าเริ่มต้นแทน เนื่องจากวิธีการทำงานของวัตถุ Python จึงควรตั้งค่าเหล่านี้ไว้ภายในฟังก์ชันไม่ใช่ในอาร์กิวเมนต์

นี่คือฟังก์ชั่นการทำงาน:

def list_flatten(l, a=None):
    #check a
    if a is None:
        #initialize with empty list
        a = []

    for i in l:
        if isinstance(i, list):
            list_flatten(i, a)
        else:
            a.append(i)
    return a

การทดสอบ:

In [2]: lst = [1, 2, [3], [[4]],[5,[6]]]

In [3]: lst
Out[3]: [1, 2, [3], [[4]], [5, [6]]]

In [11]: list_flatten(lst)
Out[11]: [1, 2, 3, 4, 5, 6]

ต่อไปนี้ดูเหมือนง่ายที่สุดสำหรับฉัน:

>>> import numpy as np
>>> l = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
>>> print (np.concatenate(l))
[1 2 3 4 5 6 7 8 9]

ท่านสามารถใช้แฟลตของ NumPy :

import numpy as np
list(np.array(l).flat)

แก้ไข 11/02/2016: ใช้ได้เฉพาะเมื่อรายการย่อยมีขนาดเท่ากัน

คุณสามารถใช้จำนวน:
flat_list = list(np.concatenate(list_of_list))

หากคุณยินดีที่จะละทิ้งความเร็วเล็กน้อยเพื่อให้ดูสะอาดตาคุณสามารถใช้numpy.concatenate().tolist()หรือnumpy.concatenate().ravel().tolist():

import numpy

l = [[1, 2, 3], [4, 5, 6], [7], [8, 9]] * 99

%timeit numpy.concatenate(l).ravel().tolist()
1000 loops, best of 3: 313 µs per loop

%timeit numpy.concatenate(l).tolist()
1000 loops, best of 3: 312 µs per loop

%timeit [item for sublist in l for item in sublist]
1000 loops, best of 3: 31.5 µs per loop

คุณสามารถหาข้อมูลเพิ่มเติมได้ที่นี่ในเอกสารnumpy.concatenateและnumpy.ravel

ฉันพบวิธีแก้ปัญหาที่เร็วที่สุด (สำหรับรายการขนาดใหญ่แล้ว):

import numpy as np
#turn list into an array and flatten()
np.array(l).flatten()

ทำ! แน่นอนว่าคุณสามารถเปลี่ยนกลับเป็นรายการได้โดยดำเนินการรายการ (l)

รหัสง่าย ๆ สำหรับunderscore.pyแฟนแพ็คเกจ

from underscore import _
_.flatten([[1, 2, 3], [4, 5, 6], [7], [8, 9]])
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

มันแก้ปัญหาแผ่แบนทั้งหมด (ไม่มีรายการในรายการหรือการซ้อนที่ซับซ้อน)

from underscore import _
# 1 is none list item
# [2, [3]] is complex nesting
_.flatten([1, [2, [3]], [4, 5, 6], [7], [8, 9]])
# [1, 2, 3, 4, 5, 6, 7, 8, 9]

คุณสามารถติดตั้งunderscore.pyด้วย pip

pip install underscore.py

def flatten(alist):
    if alist == []:
        return []
    elif type(alist) is not list:
        return [alist]
    else:
        return flatten(alist[0]) + flatten(alist[1:])

หมายเหตุ : ด้านล่างนี้นำไปใช้กับงูหลาม 3.3 yield_fromหรือสูงกว่าเพราะมันใช้ sixเป็นแพ็คเกจของบุคคลที่สามแม้ว่าจะมีเสถียรภาพ sys.versionหรือคุณสามารถใช้

ในกรณีที่มีทั้งหมดของโซลูชั่นที่นี่เป็นสิ่งที่ดีรวมทั้งความเข้าใจและรายการobj = [[1, 2,], [3, 4], [5, 6]]itertools.chain.from_iterable

อย่างไรก็ตามให้พิจารณากรณีที่ซับซ้อนเล็กน้อยนี้:

>>> obj = [[1, 2, 3], [4, 5], 6, 'abc', [7], [8, [9, 10]]]

มีปัญหาหลายอย่างที่นี่:

• องค์ประกอบหนึ่ง6, เป็นเพียงสเกลาร์; ไม่สามารถทำซ้ำได้ดังนั้นเส้นทางด้านบนจะล้มเหลวที่นี่
• องค์ประกอบหนึ่ง'abc', เป็นเทคนิค iterable (ทุกstrs) อย่างไรก็ตามการอ่านระหว่างบรรทัดเล็กน้อยคุณไม่ต้องการที่จะปฏิบัติกับมันเช่น - คุณต้องการที่จะปฏิบัติกับมันเป็นองค์ประกอบเดียว
• องค์ประกอบสุดท้าย[8, [9, 10]]คือตัวเอง iterable ซ้อนกัน ความเข้าใจพื้นฐานของรายการและchain.from_iterableเพียงดึง "ลง 1 ระดับ" เท่านั้น

คุณสามารถแก้ไขได้ดังนี้:

>>> from collections import Iterable
>>> from six import string_types

>>> def flatten(obj):
...     for i in obj:
...         if isinstance(i, Iterable) and not isinstance(i, string_types):
...             yield from flatten(i)
...         else:
...             yield i


>>> list(flatten(obj))
[1, 2, 3, 4, 5, 6, 'abc', 7, 8, 9, 10]

ที่นี่คุณตรวจสอบว่าองค์ประกอบย่อย (1) สามารถทำซ้ำได้ด้วยIterableABC จากitertoolsแต่ยังต้องการให้แน่ใจว่า (2) องค์ประกอบนั้นไม่ใช่ "เหมือนสตริง"

flat_list = []
for i in list_of_list:
    flat_list+=i

จรรยาบรรณนี้ยังใช้งานได้ดีเพราะมันขยายรายการไปเรื่อย ๆ แม้ว่ามันจะคล้ายกันมาก แต่มีเพียงหนึ่งวงเท่านั้น ดังนั้นจึงมีความซับซ้อนน้อยกว่าการเพิ่ม 2 สำหรับลูป

from nltk import flatten

l = [[1, 2, 3], [4, 5, 6], [7], [8, 9]]
flatten(l)

ข้อดีของการแก้ปัญหานี้เหนือคนอื่นส่วนใหญ่ที่นี่คือถ้าคุณมีรายการเช่น:

l = [1, [2, 3], [4, 5, 6], [7], [8, 9]]

ในขณะที่โซลูชันอื่น ๆ ส่วนใหญ่มีข้อผิดพลาดโซลูชันนี้จะจัดการกับพวกเขา

นี่อาจไม่ใช่วิธีที่มีประสิทธิภาพที่สุด แต่ฉันคิดว่าจะใช้หนึ่งซับ (จริง ๆ แล้วเป็นสองซับ) ทั้งสองเวอร์ชันจะทำงานกับรายการที่ซ้อนกันตามลำดับชั้นโดยพลการและใช้ประโยชน์จากคุณลักษณะทางภาษา (Python3.5) และการเรียกซ้ำ

def make_list_flat (l):
    flist = []
    flist.extend ([l]) if (type (l) is not list) else [flist.extend (make_list_flat (e)) for e in l]
    return flist

a = [[1, 2], [[[[3, 4, 5], 6]]], 7, [8, [9, [10, 11], 12, [13, 14, [15, [[16, 17], 18]]]]]]
flist = make_list_flat(a)
print (flist)

ผลลัพธ์คือ

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]

วิธีนี้ใช้ได้ผลในเชิงลึกก่อน การเรียกซ้ำลงไปจนกระทั่งพบองค์ประกอบที่ไม่ใช่รายการจากนั้นขยายตัวแปรโลคัลflistและจากนั้นย้อนกลับไปยังพาเรนต์ เมื่อใดก็ตามที่flistถูกส่งคืนจะถูกขยายไปยังผู้ปกครองflistในรายการความเข้าใจ ดังนั้นที่รูทรายการแฟล็กจะถูกส่งคืน

รายการด้านบนสร้างรายการท้องถิ่นหลายรายการและส่งคืนรายการที่ใช้เพื่อขยายรายการผู้ปกครอง ฉันคิดว่าวิธีการสำหรับสิ่งนี้อาจเป็นการสร้างภาพลักษณ์ที่เยือกเย็นflistดังด้านล่าง

a = [[1, 2], [[[[3, 4, 5], 6]]], 7, [8, [9, [10, 11], 12, [13, 14, [15, [[16, 17], 18]]]]]]
flist = []
def make_list_flat (l):
    flist.extend ([l]) if (type (l) is not list) else [make_list_flat (e) for e in l]

make_list_flat(a)
print (flist)

ผลลัพธ์เป็นอีกครั้ง

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18]

แม้ว่าฉันไม่แน่ใจในเวลานี้เกี่ยวกับประสิทธิภาพ

อีกวิธีที่ผิดปกติที่ใช้กับรายการจำนวนเต็มและเป็นเนื้อเดียวกัน:

from typing import List


def flatten(l: list) -> List[int]:
    """Flatten an arbitrary deep nested list of lists of integers.

    Examples:
        >>> flatten([1, 2, [1, [10]]])
        [1, 2, 1, 10]

    Args:
        l: Union[l, Union[int, List[int]]

    Returns:
        Flatted list of integer
    """
    return [int(i.strip('[ ]')) for i in str(l).split(',')]