从多个列表创建 DataFrame？

Question

我有两个列表list1=['a','b','c']list2=[1,2]我希望我的数据框输出看起来像：col1     col2a        1a        2b        1b        2c        1c        2如何才能做到这一点？

Answer 1

使用itertools.product：

import itertools

list1 = ['a','b','c']

list2 = [1,2]

df = pd.DataFrame(itertools.product(list1, list2), columns=['col1', 'col2'])

print(df)

输出：

  col1  col2

0    a     1

1    a     2

2    b     1

3    b     2

4    c     1

5    c     2

Answer 2

如果您不想显式import itertools,pd.MultiIndex有一个from_product方法可以搭载：

list1 = ['a','b','c']

list2 = [1, 2]

pd.DataFrame(pd.MultiIndex.from_product((list1, list2)).to_list(), columns=['col1', 'col2'])

  col1  col2

0    a     1

1    a     2

2    b     1

3    b     2

4    c     1

5    c     2