python保存高维数组array

  1. 如果是1维或2维数组,可以使用np.savetxt(), 如:

    import numpy as np
    
    # 创建数组(2维)
    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=",")
    
  2. 如果是高维数组,可以使用np.save(),如:

    import numpy as np
    
    # 创建数组(3维)
    a = np.arange(100).reshape((10, 5, 2))
    
    # 存储
    np.save(file="data.npy", arr=a)
    
    # 读取
    b= np.load(file="data.npy")
    

你可能感兴趣的:(python保存高维数组array)