python - 向 DataFrame 添加多行的有效方法

Question

我真的想加快我的代码速度。

我已经工作的代码循环遍历 DataFrame 并获取开始和结束年份。然后我将其添加到列表中。在循环结束时，我 append 到空的 DataFrame。

rows = range(3560)

#initiate lists and dataframe
start_year = []
end_year = []

for i in rows:

    start_year.append(i)
    end_year.append(i)


df = pd.DataFrame({'Start date':start_year, 'End date':end_year})

我得到了我所期望的结果，但速度非常慢:

    Start date  End date
0   1            1
1   2            2
2   3            3
3   4            4

Answer 1

是的，它可以做得更快。诀窍是避免 list.append (或者，更糟糕的是 pd.DataFrame.append )在循环中。您可以使用list(range(3560))，但您可能会发现np.arange甚至更加高效。在这里您可以通过dict.fromkeys将一个数组分配给多个系列:

df = pd.DataFrame(dict.fromkeys(['Start date', 'End date'], np.arange(3560)))

print(df.shape)
# (3560, 2)

print(df.head())
#    Start date  End date
# 0           0         0
# 1           1         1
# 2           2         2
# 3           3         3
# 4           4         4