[记录点滴] 一个Python中实现flatten的方法

2021-05-03 19:29

阅读:731

标签:tool   python   imp   import   aaa   列表   去重   chain   一个   

之前如果想使用flatten,一般借助于numpy.ndarray.flatten。

但是 flatten只能适用于numpy对象,即array或者mat,普通的list列表不适用。

最近找到一个轻便的办法如下:

from itertools import chain

# flatten
print(list(set(chain.from_iterable(["aaa", "bbb", ["c","d", "e"]]))))

# 输出[‘b‘, ‘d‘, ‘c‘, ‘a‘, ‘e‘]

如果不想去重,则类似思路也可以将嵌套列表转换成单一列表:

print(list(chain.from_iterable([[1, 3], [4, 4], [5,8,7]])))

# 输出[1, 3, 4, 4, 5, 8, 7]

[记录点滴] 一个Python中实现flatten的方法

标签:tool   python   imp   import   aaa   列表   去重   chain   一个   

原文地址:https://www.cnblogs.com/rossiXYZ/p/13172139.html


评论


亲,登录后才可以留言!