如何在python中保存2D numpy数组而不是行方式?

时间:2016-12-10 08:32:45

标签: python csv pandas numpy

我有这个数组:

[[1,2,3],[4,5],[7,8,9]]

我尝试使用pandas.to_csv,但它给了我

1 2 3
4 5 nan
7 8 9

但我想要

1 4 7
2 5 8 
3 nan 9

1 个答案:

答案 0 :(得分:0)

我首先要确保所有行都具有相同的长度:

a = [[], [1], [1,2], [1,2,3]]
max_row_length = max([len(r) for r in a])
a = [r + [None]*(max_row_length - len(r)) for r in a]

然后我会使用numpy:

np.array(a).T