Python：有效的展开/扁平列表和单个浮子的列表

Question

我有一个元组和单个浮子的列表，

v = [(1., 2., 3.), (4., 5.), 6., 7.]

现在我想用良好的旧列表理解来将其弄平，

[item for sublist in v for item in sublist]

但是由于单浮子是不可能的，因此发生了错误。展开此混合列表的最有效方法是什么？我尝试浏览列表并使用isininstance（），但再次，列表是不可能的。

编辑：到目前为止

• 使用HSTACK的时间比较：6.49 µs±100 ns / loop
• hasattr方法：398 ns±2.11 ns / loop
• Collections方法：804 ns±1.27 ns / loop

Answer 1

您可以使用 hasattr该对象是可以触及的（具有__ iTer __方法）：

out = [item for sublist in v
       for item in (sublist if hasattr(sublist, '__iter__') else [sublist])]

输出：[1.0，2.0，3.0，4.0，5.0，6.0，6.0，7.0]

替代方法是使用 collect> collections.abc ：

from collections.abc import Iterable
    
out = [item for sublist in v
       for item in (sublist if isinstance(sublist, Iterable) else [sublist])]   

在这两种情况下，字符串都是可观的，您可能需要根据数据检查此特定情况。

Answer 2

使用numpy您可以使用 hstack（） /a>：

v = [(1., 2., 3.), (4., 5.), 6., 7.]
v = np.hstack(v)

# Output
array([1., 2., 3., 4., 5., 6., 7.])

Answer 3

more-itertools package 软件包具有专门用于此问题的函数

from more_itertools import collapse
v = [(1., 2., 3.), (4., 5.), 6., 7.]
print(list(collapse(v))
[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0]

-也可以使用字符串 -

x = [('ab', 'cd'), 'e', ['f', 'gh']]
print(list(collapse(x))
['ab', 'cd', 'e', 'f', 'gh']