python - Pandas 将数据框列单元格初始化为空列表

Question

我需要将 DataFrame 的列中的单元格初始化为 lists。

df['some_col'] = [[] for _ in no_of_rows]

我想知道在时间效率方面是否有更好的方法来做到这一点？

Answer 1

由于您正在寻找时间效率，因此低于一些基准。我认为 list 理解已经相当快地创建 list 对象的空 list ，但是你可以使用 挤出边际改进itertools.repeat。在 insert 部分，apply 慢了 3 倍，因为它循环:

import pandas as pd
from itertools import repeat
df = pd.DataFrame({"A":np.arange(100000)})

%timeit df['some_col'] = [[] for _ in range(len(df))]
100 loops, best of 3: 8.75 ms per loop

%timeit df['some_col'] = [[] for i in repeat(None, len(df))]
100 loops, best of 3: 8.02 ms per loop

%%timeit 
df['some_col'] = ''
df['some_col'] = df['some_col'].apply(list)
10 loops, best of 3: 25 ms per loop