所以我有一个如下列表:
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.