如何在Python中读取数据集?
时间: 2024-05-13 13:11:17 浏览: 131
为了读取数据集,可以使用Python中的多种库和函数,其中包括:
1.使用pandas库读取csv文件:
```python
import pandas as pd
dataset = pd.read_csv('file_path.csv')
```
2.使用numpy库读取txt文件:
```python
import numpy as np
dataset = np.loadtxt('file_path.txt')
```
3.使用scipy库读取mat文件:
```python
import scipy.io as sio
dataset = sio.loadmat('file_path.mat')
```
4.使用sklearn库读取自带的数据集:
```python
from sklearn.datasets import load_iris
iris = load_iris()
```
5.使用torch.utils.data库读取自定义数据集:
```python
from torch.utils.data import Dataset, DataLoader
class MyDataset(Dataset):
def __init__(self, data_file):
self.data = pd.read_csv(data_file)
def __len__(self):
return len(self.data)
def __getitem__(self, idx):
return self.data.iloc[idx, :-1].values, self.data.iloc[idx, -1]
dataset = MyDataset('file_path.csv')
```
阅读全文