在python中将数组列表和列表导出到csv文件

Question

我找到了几个帮助用户在 python 中将列表写入 csv 文件的链接。但是，我找不到我正在使用的特定列表格式的解决方案。我有一个列表，其中每个索引都采用以下形式：[array([x1,x2,x3,x4]),[x5,x6]] 我的列表有数千个索引。 我想将此列表导出到 csv 文件，其中 x1 是第一列，x2 是第二列，直到 x6 作为第六列。 我已经尝试过如下熊猫库。但是，这会将第一列中的数组和列表转储到另一列，因此只提供两列而不是我想要的六列。

td_df = pd.DataFrame(training_data_arr)
td_df.to_csv('DataSet.csv', header=False, index=False)

此外，我不希望我的数据在每次运行代码时都被覆盖。相反，我希望我的数据从 csv 文件中的下一个空行写入

编辑：我的数据看起来像

[[[0.004, 0.336, 0.009, -0.212],[1,0]], [[0.24, 0.66, 0.09, -0.2],[1,0]], [[-0.11, 0.16, 0.77, -0.2],[0,1]]...]

Answer 1

要附加到 csv 文件，请自己打开文件并将其传递给to_csv，例如：

with open('CartPoleDataSet.csv', 'a') as f:
    td_df.to_csv(f, header=False, index=False)

听起来您只是想在导出数据之前将其展平。

一种扁平化方法是：

In []
pd.DataFrame(td_df.sum(axis=1).values.tolist())

Out[]:
      0     1     2      3  4  5
0 0.004 0.336 0.009 -0.212  1  0
1 0.240 0.660 0.090 -0.200  1  0