python - 从不等长的嵌套列表创建 Pandas 数据框

Question

所以我有一个列表如下:

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'])

但结果是，每个列表都被写成一行(而不是一列)

Answer 1

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。