用python实现读取csv数据,并每50行保存第三、四、五、六、七列为一个mat文件
时间: 2024-01-26 14:04:31 浏览: 137
import pandas as pd
import numpy as np
import scipy.io as sio
# 读取csv文件
df = pd.read_csv('data.csv')
# 每50行保存为一个mat文件
for i in range(0, len(df), 50):
data = df.iloc[i:i+50, 2:7].values
filename = 'data_'+str(i//50)+'.mat'
sio.savemat(filename, {'data': data})
相关问题
用python实现读取csv数,并每50行保存为一个mat文件
下面是一个实现读取csv并每50行保存为一个mat文件的Python脚本:
```python
import csv
import numpy as np
from scipy.io import savemat
# 读取csv文件
with open('data.csv', 'r') as f:
reader = csv.reader(f)
# 跳过表头
next(reader)
# 初始化mat文件计数器
mat_count = 0
# 初始化mat文件数据
mat_data = []
for i, row in enumerate(reader):
# 将csv行数据转换为numpy数组
row_data = np.array(row, dtype=np.float32)
# 添加到mat文件数据中
mat_data.append(row_data)
# 每50行保存一次mat文件
if (i+1) % 50 == 0:
# 构造mat文件名
mat_name = 'data_{}.mat'.format(mat_count)
# 保存mat文件
savemat(mat_name, {'data': np.array(mat_data)})
# 清空mat文件数据
mat_data = []
# 更新mat文件计数器
mat_count += 1
# 保存最后一个mat文件
if len(mat_data) > 0:
mat_name = 'data_{}.mat'.format(mat_count)
savemat(mat_name, {'data': np.array(mat_data)})
```
其中,`data.csv`为要读取的csv文件名,每50行数据将保存为一个名为`data_0.mat`、`data_1.mat`、`data_2.mat`等的mat文件。`savemat`函数用于将numpy数组保存为mat文件,保存的变量名为`data`。
写一段python代码,要求:读取csv文件,将文件的第3列到第12列的数据保存到data.mat文件,第16列数据保存到label.mat文件中
以下是符合要求的 Python 代码:
```python
import numpy as np
import pandas as pd
# 读取 CSV 文件
data = pd.read_csv('data.csv')
# 提取指定列的数据
data_features = data.iloc[:, 2:12].values
data_label = data.iloc[:, 15:16].values
# 保存为 mat 文件
np.save('data.mat', data_features)
np.save('label.mat', data_label)
```
注:上述代码中,需将 `data.csv` 文件保存在当前工作目录中。此外,需要安装 `numpy` 和 `pandas` 库。
阅读全文