在不更改python中的外部列表的情况下展平内部列表

时间:2013-04-20 21:26:54

标签: python list nested flatten

因此,只需在python flatten list中键入stack-overflow / google就会带来大量重复的结果。我已经尝试了所有这些,但没有一个适用于这个特定的例子。

我从pyodbc返回一个数据库结果集,它给了我一个数据结构,我可以强制到列表列表。返回的每一行中的第一个元素(第一列)采用"a.b.c"形式。我希望它是"a", "b", "c"。我的第一直觉是分裂时期,我做了。

在:

[["a.b.c", "d", 1], ["e.f.g", "h", 2], ... ] # rows left out for brevity

后:

# unfortunately, split returns a list, which then gets nested
[[["a", "b", "c"], "d", 1], [["e", "f", "g"], "h", 2], ... ]

但是,我想看到的是:

[["a", "b", "c", "d", 1], ["e", "f", "g", "h", 2], ... ]

我尝试使用堆栈溢出的先前解决方案来展平列表,但是当每个人都提到嵌套列表时,没有人说如何展平嵌套列表。

我试过了:

from itertools import chain
for row in rows:
    row = list(chain(row)) # python won't allow modifications in a loop

rows = [list(chain(row)) for row in rows]

我认为必须有一种方式,或许有收益率,做一些像:

for row in rows:
    row = row[0].append(row[1:]) # but this doesn't work either

我不知道如何修改列表列表中的内部列表。如果有比我迄今为止尝试过的更好的方式,我很乐意听到它。在此先感谢您的帮助。

1 个答案:

答案 0 :(得分:7)

怎么样:

>>> s = [["a.b.c", "d", 1], ["e.f.g", "h", 2]]
>>> [k[0].split('.') + k[1:] for k in s]
[['a', 'b', 'c', 'd', 1], ['e', 'f', 'g', 'h', 2]]

这将采用split在对第一个元素进行操作后返回的列表,并将其添加到由其余元素组成的列表中。