通过连接2个列表(如2列的乘法)创建新的数据帧 - Python

Question

问题可能从标题中不清楚,因为我不确定该怎么称呼它.我想在Pandas中实现以下目标:

a = [1,2,3,4] b = ['a','b']

我可以创建一个DataFrame:

column1 column2
'a'        1
'a'        2
'a'        3
'a'        4
'b'        1
'b'        2
'b'        3
'b'        4

Answer 1

itertools.product与DataFrame构造函数一起使用:

a = [1,2,3,4] 
b = ['a', 'b']

from  itertools import product
#pandas 0.24.0+
df = pd.DataFrame(product(b,a), columns=['column1','column2'])
#pandas bellow 
#df = pd.DataFrame(list(product(b,a)), columns=['column1','column2'])
print (df)
  column1  column2
0       a        1
1       a        2
2       a        3
3       a        4
4       b        1
5       b        2
6       b        3
7       b        4