gpt4 book ai didi

Python:元组和单个 float 的高效展开/展平列表

转载 作者:行者123 更新时间:2023-12-05 09:02:02 25 4
gpt4 key购买 nike

我有一个元组列表和单个 float

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

现在我想用良好的旧列表理解来展平它

[item for sublist in v for item in sublist]

但由于单个 float 不可迭代,因此会发生错误。展开此混合列表的最有效方法是什么?我尝试遍历列表并使用 isininstance() 但再一次,列表不可迭代。

编辑:到目前为止的时间比较

  • 使用 hstack 的 numpy 方法:每个循环 6.49 µs ± 100 ns
  • hasattr 方法:每个循环 398 ns ± 2.11 ns
  • 收集方法:每个循环 804 ns ± 1.27 ns

最佳答案

您可以使用 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, 7.0]

另一种方法是使用 collections.abc :

from collections.abc import Iterable

out = [item for sublist in v
for item in (sublist if isinstance(sublist, Iterable) else [sublist])]

请注意,在这两种情况下,字符串都被认为是可迭代的,您可能需要根据您的数据检查这种特定情况。

关于Python:元组和单个 float 的高效展开/展平列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/72197341/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com