用python实现读取csv数,并每50行保存为一个mat文件
时间: 2024-01-13 22:04:31 浏览: 86
下面是一个实现读取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`。
阅读全文