Python，如何將不等長的嵌套列表寫入csv文件？

Question

假設我有一個結構如下的numpy數組：

[['a','b','c'],[1,2,3],['i','j','k','l'],[5,10,15,20]]

我想將它保存到一個看起來像這樣的csv文件

a, 1, i, 5
b, 2, j, 10
c, 3, k, 15
,  ,  l, 20

長度較短的列只需填空。 我怎樣才能做到這一點？

Answer 1

使用itertools.izip_longest ：

>>> from itertools import izip_longest
>>> lis =  [['a','b','c'],[1,2,3],['i','j','k','l'],[5,10,15,20]]
>>> list(izip_longest(*lis, fillvalue=''))
[('a', 1, 'i', 5),
 ('b', 2, 'j', 10),
 ('c', 3, 'k', 15),
 ('', '', 'l', 20)]

使用csv.writerows(izip_longest(*lis, fillvalue=''))將其寫入csv文件。