python – 从不等长度的嵌套列表创建一个pandas数据帧

所以我有一个如下列表：

aa = ['aa1', 'aa2', 'aa3', 'aa4', 'aa5']
bb = ['bb1', 'bb2', 'bb3', 'bb4']
cc = ['cc1', 'cc2', 'cc3']

然后将其创建为嵌套列表：

nest = [aa, bb, cc]

我想创建一个数据帧,如下所示：

aa   bb   cc
aa1  bb1  cc1
aa2  bb2  cc2
aa3  bb3  cc3
aa4  bb4  nan
aa5  nan  nan

我试过了：

pd.DataFrame(nest, columns=['aa', 'bb', cc'])

但结果就是这样,每个列表都被写成一行(而不是列)

解决方法:

itertools的zip_longest函数执行此操作：

>>> import itertools, pandas
>>> pandas.DataFrame((_ for _ in itertools.zip_longest(*nest)), columns=['aa', 'bb', 'cc'])
    aa    bb    cc
0  aa1   bb1   cc1
1  aa2   bb2   cc2
2  aa3   bb3   cc3
3  aa4   bb4  None
4  aa5  None  None

如果您有旧版本的pandas,则可能需要在列表构造函数中包装zip_longest.在较旧的Python上,您可能需要调用izip_longest而不是zip_longest.