如何将pandas的dateframe格式转化为torch.tensor格式
时间: 2023-06-01 17:05:35 浏览: 210
PandasDataFrame-1.0.2.tar.gz
可以使用pandas.DataFrame.values方法将DataFrame转换为numpy.ndarray,然后使用torch.tensor方法将numpy.ndarray转换为torch.tensor格式。具体代码可参考以下示例:
```
import pandas as pd
import torch
# 创建一个DataFrame
df = pd.DataFrame({'a': [1, 2, 3], 'b': [4, 5, 6]})
# 转换为torch.tensor格式
tensor = torch.tensor(df.values)
```
阅读全文