如何使用 Pandas 保存 N x M 数组/列表？

Posted 2023-05-07

【中文标题】如何使用 Pandas 保存 N x M 数组/列表？

【英文标题】：How do I save a N x M array/list using Pandas?

【发布时间】：2021-11-03 17:41:37

【问题描述】：

我有一个N x M numpy 数组/列表。我想使用 Pandas 将此矩阵保存到 .csv 文件中。不幸的是，我先验不知道 M 和 N 的值可能是 大。我对 Pandas 很感兴趣，因为我发现它在数据列访问方面易于管理。

让我们从这个 MWE 开始：

import numpy as np
import pandas as pd

N,M = np.random.randint(10,100, size = 2)
A = np.random.randint(10, size = (N,M))

columns = []

for i in range(len(A[0,:])):
   columns.append( "column_ ".format(i)  )

我不能做类似pd.append( ) 的事情，即通过 for 循环附加具有新的附加索引的列。

有没有办法将 A 保存到 .csv 文件中？

【问题讨论】：

pd.DataFrame(A).to_csv('yourfile.csv')?

另外，你为什么不用np.save/load？

@QuangHoang 因为我不知道这些命令的存在。请将此/这些评论重新排列为答案。

【参考方案1】：

根据 Quang Hoang 的评论，有两种可能：

pd.DataFrame(A).to_csv('yourfile.csv')。 np.save("yourfile.npy",A)，然后是A = np.load("yourfile.npy")。