首页 > 解决方案 > 将不同的数组保存到一个 csv 文件中

问题描述

我有一个数组,scores其值如下

[[0.66372503]
 [0.64839758]
 [0.63307013]]

然后,我有一个数组,scores2其值如下

[[0.65367322]
 [0.63598164]
 [0.62295124]]

如何将它们组合并保存到 csv 中

         scores         scores2              
1     0.66372503     0.65367322
2     0.64839758     0.63598164
3     0.63307013     0.62295124

? 谢谢

标签: arrayspandascsv

解决方案


df= pd.DataFrame(data={'scores': [0.66372503, 0.64839758, 0.63307013],
                       'scores2':[0.65367322, 0.63598164, 0.62295124]})
df.to_csv('filename.csv', index=False)

如果您有 2d 数组,则可以在将数组传递给 pandas df 之前使用 numpy flatten 方法:

scores= [[0.66372503], [0.64839758], [0.63307013]]
scores2=[[0.65367322], [0.63598164], [0.62295124]]

df= pd.DataFrame(data={'scores': np.array(scores).flatten(),
                       'scores2':np.array(scores2).flatten()})

df.to_csv('filename.csv', index=False)

推荐阅读