Python数据分析与展示:ndarray多维数组的存储与读取-2

csv文件

np.savetxt()
np.loadtxt()

多维数组

a.tofile()
np.fromfile()
np.save()
np.savez()
np.load()

代码实例

# -*- coding: utf-8 -*-
# @File    : csv_demo.py
# @Date    : 2018-05-06
import numpy as np
# 存贮和读取1维或2维数组
def foo1():
    a = np.arange(100).reshape((20, 5))
    # 写入文件
    np.savetxt(fname="data.csv", X=a, fmt="%d",delimiter=",")
    # 读取文件
    b = np.loadtxt(fname="data.csv", dtype=np.int, delimiter=",")
    print(b)
# 写入读取多维数组数组结构会丢失
def foo2():
    # 可以将数据结构存入一个文件,读取
    a = np.arange(100).reshape((10, 5, 2))
    # 写入文件
    a.tofile(file="data.dat", sep=",", format="%s")
    # 读取文件
    b = (np.fromfile(file="data.dat", dtype=np.int, count=-1, sep=",")
            .reshape((10, 5, 2)))
    print(b)
# 便捷的存取读出npy文件
def foo3():
    a = np.arange(100).reshape((10, 5, 2))
    # 存储
    np.save(file="data.npy", arr=a)
    # 读取
    b= np.load(file="data.npy")
    print(b)