python讀取、保存npy文件

---

文章目錄

• python讀取、保存npy文件
• • 1 讀取與保存npy文件
  • 2 示例：將npy轉(zhuǎn)化為csv

npy文件是numpy專用的二進(jìn)制文件

這里簡單展示：

• 如何利用python讀取、保存.npy格式文件
• 將.npy轉(zhuǎn)化為.csv

1 讀取與保存npy文件

import numpy as npdata = np.zeros((3,4)) np.save('data.npy', data) del data data = np.load('data.npy')print('type :', type(data)) print('shape :', data.shape) print('data :') print(data)

輸出如下：

type : <class 'numpy.ndarray'> shape : (8, 4) data : [[0. 0. 0. 0.][0. 0. 0. 0.][0. 0. 0. 0.]]

2 示例：將npy轉(zhuǎn)化為csv

具體要求如下：

加載npy格式的數(shù)據(jù)文件

給每一列數(shù)據(jù)添加一個列名

以csv格式保存數(shù)據(jù)

import pandas as pd import numpy as np import csvdata = np.load('data.npy') columns = ['A', 'B', 'C', 'D'] with open('data.csv', 'w') as f:csv_writer = csv.writer(f)csv_writer.writerow(columns)csv_writer.writerows(data) # 或者 # ------------------------------ # # data = np.load('data.npy') # columns = np.array([['A', 'B', 'C', 'D']]) # data = np.concatenate((columns, data), axis=0) # with open('data.csv', 'w') as f: # csv_writer = csv.writer(f) # csv_writer.writerows(data) # ------------------------------ # df = pd.read_csv('data.csv') print(df)

輸出如下：

A B C D 0 0.0 0.0 0.0 0.0 1 0.0 0.0 0.0 0.0 2 0.0 0.0 0.0 0.0

注意：

• writerow與writerows的區(qū)別

• 第二種方法中，columns的類型應(yīng)該是二維numpy數(shù)組，而不是列表。否則在與data合并時，會出現(xiàn)維度錯誤。

總結(jié)

以上是生活随笔為你收集整理的python读取、保存npy文件的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。