我需要将DataFrame列中的单元格初始化为列表.
df['some_col'] = [[] for _ in no_of_rows]
我想知道在时间效率方面有更好的方法吗?
解决方法:
因为您正在寻找时间效率,低于一些基准.我认为列表理解已经非常快,可以创建列表对象的空列表,但是您可以使用itertools.repeat挤出一个边际改进.在插入件上,应用速度慢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