pandas将dataframe列单元格初始化为空列表

Question

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

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

我想知道在时间效率方面有更好的方法吗？

Answer 1

试试apply ：

df1['some_col'] = ''
df1['some_col'] = df1['some_col'].apply(list)

样品：

df1 = pd.DataFrame({'a': pd.Series([1,2])})
print (df1)
   a
0  1
1  2

df1['some_col'] = ''
df1['some_col'] = df1['some_col'].apply(list)
print (df1)
   a some_col
0  1       []
1  2       []

Answer 2

因为您正在寻找时间效率，低于一些基准。 我认为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