我有一長串由 numpy 數組和整數組成的復雜列表，下面是一個示例：[array([[2218.67288865]]), array([[1736.90215229]]), array([[1255.13141592]]), array([[773.36067956]]), array([[291.58994319]]), 0, 0, 0, 0, 0, 0, 0, 0, 0]我想將其轉換為常規列表：[2218.67288865, 1736.90215229, 1255.13141592, 773.36067956, 291.58994319, 0, 0, 0, 0, 0, 0, 0, 0, 0]我怎樣才能有效地做到這一點？
3 回答
函數式編程
TA貢獻1807條經驗 獲得超9個贊
這里最簡單的似乎也是最快的：
x = [array([[2218.67288865]]), array([[1736.90215229]]), array([[1255.13141592]]), array([[773.36067956]]), array([[291.58994319]]), 0, 0, 0, 0, 0, 0, 0, 0, 0]
[y if y.__class__==int else y.item(0) for y in x]
# [2218.67288865, 1736.90215229, 1255.13141592, 773.36067956, 291.58994319, 0, 0, 0, 0, 0, 0, 0, 0, 0]
timeit(lambda:[y if y.__class__==int else y.item(0) for y in x])
# 2.198630048893392
寶慕林4294392
TA貢獻2021條經驗 獲得超8個贊
list(itertools.from_iterable(itertools.from_iterable(...)))itertools.from_iterable(...)應該適用于刪除 2 級嵌套：只需根據需要添加或刪除副本。
牛魔王的故事
TA貢獻1830條經驗 獲得超3個贊
您可以使用生成器來展平嵌套列表：
def convert(obj):
try:
for item in obj:
yield from convert(item)
except TypeError:
yield obj
result = list(convert(data))
添加回答
舉報
0/150
提交
取消