先和大家说明一下维度转换的关系,直接用事例来演示
x=[1,2,3,4,5,6,7,8,9] # 一维列表
y=[[1,2,3],[4,5,6],[7,8,9]] # 二维列表
z=[[[1,2],3],[[4,5],6],[[7,8],9]] # 三维列表
通过这个演示应该可以直观看出列表不同维数的区别
常用的转换方式有两种,一种是利用numpy转成数组之后利用numpy的工具进行转换,另一种是通过列表的叠加进行
第一种 利用numpy
import numpy as np
x = [1, 2, 3, 4, 5, 6, 7, 8, 9] # 一维列表
Y = np.array(x).reshape(3, 3)
print(Y.reshape(3, 3))
运行结果:
利用reshape方法进行维度变换简单方便,但是只能变换规整维度,例如上一个栗子中将1*9转为3*3,而无法自定义各个维度
第二种 叠加法
s = [1, 2, 3, 4, 5, 6, 7, 8, 9] # 一维列表
# 这里按照第一个例子的三维列表进行转换,维度为3*(2+1)
x = 2 # 第三维的维度
y = 2 # 第二维的维度
z = 3 # 第一维的维度
a = [] # 存放第三维数据
b = [] # 存放第二维数据