numpy list df tensor的相互转换

每次用python处理数据时候,都要特码的搜这些数据类型之间是怎么转换的,这里我直接总结一下。

1.numpy与list之间

numpy转list

listData=numpyData.tolist()

list转numpy

numpyData=numpy.array(listData)

2.df与list之间

df转list

#一维的dataFrame
listData=dfData.values.tolist()
#某列dataFrame
listData=df['columnName'].values.tolist()

list转df

listData = [['a', '1.2', 'apple'], ['b', '2.3', 'banana'], ['c', '3.6', 'watermelon']]
df = pd.DataFrame(listData, columns=['one', 'two', 'three'])

df的结果如图

numpy list df tensor的相互转换_第1张图片

 3.df与numpy之间

df转numpy

listData = [['a', '1.2', 'apple'], ['b', '2.3', 'banana'], ['c', '3.6', 'watermelon']]
df = pd.DataFrame(listData, columns=['one', 'two', 'three'])
#将df转为numpy
numpyData=df.to_numpy()
print(type(numpyData))
print(numpyData.shape)

 输出结果

numpy list df tensor的相互转换_第2张图片

 4.numpy与tensor之间

numpy转tensor

#第一种方法
tensorData=torch.tensor(numpyData)
#第二种方法
tensorData=torch.from_numpy(numpyData)

 tensor转numpy

numpyData = tensorData.numpy()

常用的数据类型基本就这几个,剩余的数据类型按照以上公式转换即可。

你可能感兴趣的:(AI,Python,pytorch,pandas,numpy)