python - 为列表中的值组合创建 Pandas 数据框行的最快方法

Question

假设我有三个列表

listA = ['a','b','c', 'd']
listP = ['p', 'q', 'r']
listX = ['x', 'z']

因此数据框将有 4*3*2 = 24 行。现在，解决此问题的最简单方法是:

df = pd.DataFrame(columns=['A','P','X'])

for val1 in listA:
   for val2 in listP:
      for val3 in listX:
         df.loc[<indexvalue>] = [val1,val2,val3]

现在在真实场景中，我将有大约 80 万行和 12 列(因此循环中有 12 个嵌套)。有什么办法可以更快地创建这个数据框吗？

Answer 1

类似讨论here .显然 np.meshgrid 对于大数据更有效(作为 itertools.product 的替代品。

应用:

v = np.stack(i.ravel() for i in np.meshgrid(listA, listP, listX)).T
df = pd.DataFrame(v, columns=['A', 'P', 'X'])
>>  A  P  X
0   a  p  x
1   a  p  z
2   b  p  x
3   b  p  z
4   c  p  x