将二维数组扁平化,或者说变成一维数组

发布时间 2023-07-09 19:06:05作者: 转眼春夏秋冬如烟

代码:

a = [[1, 2], [3, 4]]
# 扁平化
b = [i for item in a for i in item]
# 或
import numpy as np
npa = np.arrary(a)
b = npa.ravel()
# and
b = npa.flatten()

ravel和flatten的区别在于使用ravel形成的数据在修改后会影响np.array的源数据

上面的代码使用ravel后修改b的值npa也会变化