如何使用 Pandas 保存 N x M array/list?

How do I save a N x M array/list using Pandas?

我有一个 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 文件中?

根据Quang Hoang的评论,有2种可能:

  1. pd.DataFrame(A).to_csv('yourfile.csv').
  2. np.save("yourfile.npy",A) 然后 A = np.load("yourfile.npy").