array,list,tensor,Dataframe,Series之间互相转换总结
本文转载自:【串讲总结】array, list, tensor,Dataframe,Series之间互相转换总结
一、前言
对于在Deep Learning的学习中总会有几个数据类型的转换,这次想把这些常用的转换做一个总结,方便以后看。
这些主要包括:Dataframe、Series(pandas), array(numpy), list, tensor(torch)
二、定义
2.1 Dataframe和Series
这里简单介绍一下这两个结构。Dataframe创建的方式有很多种,这里不赘述了。以下举个例子,因为我们这里要讲的是和array等的转换,这里全都用数字型的元素。
对于dataframe来说,我们打印出来,结构类似于一个二维矩阵格式,只是每一列和每一个行都有个index,这并且这些结构之间有很多方便的操作,在读入结构化数据的时候尤为方便,所以平时做偏结构化数据的时候, 比如excel、pickle等等,pandas的使用是绕不开的。
而其中的series相当于dataframe的一个元素,如下:
Series只有row index,有点类似于一个一维向量。
而DataFrame既有行索引也有列索引,它也可以被看做由Series组成的字典(共同用一个索引)
2.2 array
数组结构是由不同维度的list转换来的,用array的原因主要在于有更多的矩阵操作,数据使用起来更方便,比如转置、矩阵相乘、reshape等等。
2.3 tensor
张量是在深度学习框架中的一个数据结构,把数据喂进模型中需要把数据转换为tensor结构,等我们再取出来做框架以外的操作,比如保存成文件,用plot画图,都需要重新转换为array或list结构。
三、互相转换
先用一个例子直观举例下
四、详细总结
4.1 Dataframe到Series
挑一列的index取出
1 | Series = Dataframe['column'] |
4.2 Series到list
1 | list = Series.to_list() |
4.3 list 转 array
1 | array = np.array(list) |
4.4 array 转 torch.Tensor
1 | tensor = torch.from_numpy(array) |
4.5 torch.Tensor 转 array
1 | array = tensor.numpy() |
4.6 torch.Tensor 转 list
1 | # 先转numpy,后转list |
4.7 array 转 list
1 | list = array.tolist() |
4.8 list 转 torch.Tensor
1 | tensor=torch.Tensor(list) |
4.9 array或者list转Series
1 | series = pd.Series({'a': array}) |
之后这里的操作就多了,看你具体需求了,也可以多个series拼成一个dataframe, 如下, 其他操作不一一赘述了
1 | df = pd.DataFrame({'aa': series, 'bb': series2}) |