python - 在 Python 中更快地遍历列表

Question

我有一个长度为 800,000 的列表 vector_list，其中元素是大小为 768 的列表。我正在尝试将 768 列添加到 pandas 数据框中，其中每列的长度为 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

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

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