脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Python读取mat文件,并保存为pickle格式

Python读取mat文件,并保存为pickle格式的方法

 更新时间：2018年10月23日 15:27:08   作者：hdc  
今天小编就为大家分享一篇Python读取mat文件,并保存为pickle格式的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

这两天在搞Theano，要把mat文件转成pickle格式载入Python。

Matlab是把一维数组当做n*1的矩阵的，但Numpy里还是有vector和matrix的区别，Theano也是对二者做了区分。

直接把代码贴出来吧，好像也没什么可讲的 = =

from scipy.io import loadmat
import numpy, cPickle
 
data_dict=loadmat(r'E:\dataset\CIFAR10\CIFAR10_small.mat') #need an r!
 
my_array=numpy.array([1,1])
for key in data_dict.keys():
 if type(data_dict[key]) == type(my_array):
  #print matrix information
  print key, type(data_dict[key]),
  print data_dict[key].shape
 
#shape(n,1) (matrix in theano) -> shape(n,) (vector in theano)
print data_dict['Ytr'].shape
Ytr=numpy.hstack(data_dict['Ytr'])
Yte=numpy.hstack(data_dict['Yte'])
Yte=numpy.hstack(data_dict['Yte'])
print Ytr.shape
 
train_set=(data_dict['Xtr'],Ytr)
valid_set =(data_dict['Xte'],Yte)
test_set =(data_dict['Xte'],Yte)
 
output = open('cifar10_small_v.pkl', 'wb')
 
cPickle.dump(train_set, output)
cPickle.dump(valid_set, output)
cPickle.dump(test_set, output)
 
output.close()
print 'save is done'
 
pkl_file = open('cifar10_small_v.pkl', 'rb')
 
data1 = cPickle.load(pkl_file) # is train_set
data2 = cPickle.load(pkl_file) # is valid_set
data3 = cPickle.load(pkl_file) # is test_set
 
print type(data1[1]),data1[1].shape
 
pkl_file.close()

以上这篇Python读取mat文件,并保存为pickle格式的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • 浅谈Python 多进程默认不能共享全局变量的问题

    浅谈Python 多进程默认不能共享全局变量的问题

    今天小编就为大家分享一篇浅谈Python 多进程默认不能共享全局变量的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-01-01
  • Python实现求两个csv文件交集的方法

    Python实现求两个csv文件交集的方法

    这篇文章主要介绍了Python实现求两个csv文件交集的方法,涉及Python针对csv文件的读取、遍历、判断等相关操作技巧,需要的朋友可以参考下
    2017-09-09
  • Python Pandas pandas.read_sql_query函数实例用法分析

    Python Pandas pandas.read_sql_query函数实例用法分析

    在本篇文章里小编给大家整理的是一篇关于Python Pandas pandas.read_sql_query函数实例用法分析内容，有兴趣的朋友们可以跟着学习下。
    2021-06-06
  • python通过cython加密代码

    python通过cython加密代码

    这篇文章主要介绍了python通过cython实现加密的示例代码，帮助大家加密自己的python代码，提高安全性，感兴趣的朋友可以参考下
    2020-12-12
  • Python变量的定义和运算符的使用

    Python变量的定义和运算符的使用

    这篇文章主要介绍了Python变量的定义和运算符的使用,Python和C/Java不同，在定义变量的时候不需要显示的指定变量的类型，在赋值的时候自动就会确定类型,需要的朋友可以参考下
    2023-05-05
  • Python构建区块链的方法详解

    Python构建区块链的方法详解

    区块链（Blockchain）是一种分布式账本（listributed ledger），它是一种仅供增加（append-only），内容不可变（immutable）的有序（ordered）链式数据结构，该数据结构由网络中的一系列节点共同维护，并且这些节点之间互不信任
    2023-02-02
  • Python Json读写操作之JsonPath用法详解

    Python Json读写操作之JsonPath用法详解

    JSONPath是一种信息抽取类库，是从JSON文档中抽取指定信息的工具，提供多种语言实现版本，包括Javascript、Python、PHP和Java，这篇文章主要介绍了Python Json读写操作之JsonPath用法详解,需要的朋友可以参考下
    2023-04-04
  • Python使用正则表达式实现爬虫数据抽取

    Python使用正则表达式实现爬虫数据抽取

    这篇文章主要介绍了Python使用正则表达式实现爬虫数据抽取，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-08-08
  • Python进程间通讯与进程池超详细讲解

    Python进程间通讯与进程池超详细讲解

    进程彼此之间互相隔离，要实现进程间通信（IPC），multiprocessing模块主要通过队列方式，队列：队列类似于一条管道，元素先进先出，需要注意的一点是：队列都是在内存中操作,进程退出,队列清空,另外,队列也是一个阻塞的形态
    2022-12-12
  • Python使用Pyqt5实现简易浏览器(最新版本测试过)

    Python使用Pyqt5实现简易浏览器(最新版本测试过)

    这篇文章主要介绍了Python使用Pyqt5实现简易浏览器(最新版本测试过)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-04-04

最新评论