Python:在 csv 中导出矩阵

Question

我有一个 13 行 13 列的二维矩阵(除第一列外还有标题)，在 Python 中名为 correl。此 correl 矩阵是从 DataFrame 生成的，我希望用多个 correl 填充矩阵 correlation。例如:

correlation=[]
correl=df.corr()
correlation=correlation.append(correl) #correlation is not a DataFrame

我之所以使用 correlation=[] 是因为我希望用多个相关表填充 correlation。这就是我使用 append 的原因，因为这是一个循环。

现在我想将这个相关矩阵导出到一个 csv 文件中。我这样做:

with open("C:\destinationfolder\file.csv", "wb") as f:
    writer = csv.writer(f)
    writer.writerows(correlation)

我收到这个错误:

raise KeyError('没有名为 %s 的项目' % com.pprint_thing(item))
KeyError: 你没有名为 0 的项目

为什么？我的猜测是我没有第一列的标题...是否有更简单的方法将我的相关表导出到 csv？

Answer 1

看起来像correlation is a DataFrame too , 所以你可以简单地使用 to_csv :

correlation.to_csv("C:\destinationfolder\file.csv")

如果您有 DataFrame 列表，请更新以回答:

如果你有一个要制作成 csv 的 DataFrame 列表，你有两个选择:

1. 将列表连接成一个更大的框架，并使用 to_csv:

   pd.concat(list_of_dataframes).to_csv(...)
   

2. 遍历每个 DataFrame 并使用带有附加 (mode='a') 的 to_csv。像这样的东西:

   list_of_dataframes[0].to_csv(...)  # write first, using headers..  assumes list isn't empty!
   for df in list_of_dataframes[1:]:
       df.to_csv(..., header=False, mode='a')
   

   参见 https://stackoverflow.com/a/17531025/1240268 .