python - 在 Python 中迭代列表的更快方法

Question

我有长度为 800,000 的列表 vector_list，其中元素是大小为 768 的列表。我正在尝试将 768 列添加到熊猫数据框中，其中每列的长度为 800,000 并代表每个列表中的一个元素。这是我的代码：

active = pd.DataFrame()    
for i in range(len(vector_list[0])):
    element_list = []
    for j in range(len(vector_list)):
        element_list.append(vector_list[j][i])
    active['Element {}'.format(i)] = element_list

只是重申一下，

len(vector_list) = 800,000
len(vector_list[0]) = 768

有没有更聪明、更快的方法来做到这一点？

标签： pythonpandasalgorithm

Answer 1

直接传递list给DataFrame构造函数。

import pandas as pd
_list = [[1, 2], [3, 4], [5, 6], [7, 8]] 
df = pd.DataFrame(_list) 
print(df.head())

输出

   0  1
0  1  2
1  3  4
2  5  6
3  7  8