python--列表、数组扁平化

列表扁平化

比如将嵌套列表（列表里含有列表）
brands=[[‘百事可乐’], [‘汇源果汁’], [‘旭日升冰茶’]]
扁平化成列表
brands1=[‘百事可乐’], [‘汇源果汁’], [‘旭日升冰茶’]

• 方法一

##利用sum()函数，命令：list1=sum(list0,[])
##sum第二个参数的默认值是0，0不能和列表相加，所以需要加入参数[]
#代码实现：
brands=[['百事可乐'], ['汇源果汁'], ['旭日升冰茶']]
brands1=sum(brands,[])
print(brands1)
#结果：['百事可乐', '汇源果汁', '旭日升冰茶']

• 方法二

#利用循环，命令：list1 = [x for j in list0 for x in j]
#代码实现：
brands1=[x for j in brands for x in j]
print(brands1)
#结果：['百事可乐', '汇源果汁', '旭日升冰茶']

• 注意

#如果列表里有各种类型的元素，比如同时包含str、列表等类型
#list = ['a', ['b'], ['c', 'd']] 
#不能使用方法一，只能使用方法二

数组扁平化

#导入模块 numpy，直接使用函数ravel()即可
import numpy as np
arr = np.array([[1,2],[3,4]])
arr.ravel()
#结果：array([1, 2, 3, 4])