pytorch代码,读取多维数据集data.csv,转换为np格式
时间: 2024-03-24 07:36:31 浏览: 57
用python读取CSV数据
以下是一个将多维CSV文件转换为NumPy数组的示例代码。假设您的CSV文件名为`data.csv`,有三列数据,第一列是输入数据1,第二列是输入数据2,第三列是目标数据:
```python
import numpy as np
import pandas as pd
# 读取CSV文件
df = pd.read_csv('data.csv')
# 将输入和目标数据分别存储在NumPy数组中
inputs_np = df.iloc[:, :2].to_numpy()
targets_np = df.iloc[:, 2].to_numpy()
```
在这个例子中,我们使用了Pandas库的`read_csv()`方法来读取CSV文件,并将其存储在名为`df`的数据框中。然后,我们使用`.iloc[]`方法将数据框中的前两列(即输入数据1和输入数据2)和第三列(即目标数据)分别存储在名为`inputs_np`和`targets_np`的NumPy数组中。现在,`inputs_np`和`targets_np`变量都包含相应的NumPy数组。
阅读全文