在python中将二维数组写入CSV文件

乌代CK

我有一个f.example返回我的函数的二维数组

array([[ 0.0, 1.0, 2.0, 3.0 ],
       [ 5.0, 1.0, 3.0, 3.0 ],
       [ 1.0, 1.0, 3.0, 3.0 ]])

事实上,我可以将它写入 csv 文件,但不是我想要的方式。以下是 csv 文件的外观:

0.0 5.0 1.0
1.0 1.0 1.0
2.0 3.0 3.0
3.0 3.0 3.0

但这就是它现在的样子:

0.0 1.0 2.0 3.0
5.0 1.0 3.0 3.0
1.0 1.0 3.0 3.0

以及我拥有的代码:

with open('example.csv', 'w') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerows(f.example)
大威

您需要先转置您的数组

import numpy as np
arr = np.array([[0.0, 1.0, 2.0, 3.0],[5.0, 1.0, 3.0, 3.0],[1.0, 1.0, 3.0, 3.0 ]])
arr_t = arr.T
print(arr_t)

输出

[[0. 5. 1.]
 [1. 1. 1.]
 [2. 3. 3.]
 [3. 3. 3.]]

然后跟之前一样。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章