2016-05-03 18 views
9

forループ内のpandas DataFrameに行を追加していますが、最後にデータフレームは常に空です。実際のforループが大量のデータを処理するため、配列に行を追加してからDataFrameコンストラクタを呼び出すことは望ましくありません。私もpd.concatを試してみました。誰かが私が陳述文を追加するために行方不明になっていることを強調できますか?ここではダミーの例です:forループ内での使用。

import pandas as pd 
import numpy as np 

data = pd.DataFrame([]) 

for i in np.arange(0, 4): 
    if i % 2 == 0: 
     data.append(pd.DataFrame({'A': i, 'B': i + 1}, index=[0]), ignore_index=True) 
    else: 
     data.append(pd.DataFrame({'A': i}, index=[0]), ignore_index=True) 

print data.head() 

Empty DataFrame 
Columns: [] 
Index: [] 
[Finished in 0.676s] 

答えて

6

あなたが付加されたデータフレームに変数data等しいを設定する必要があります。 Pythonのリストにappend方法とは異なりパンダappend

import pandas as pd 
import numpy as np 

data = pd.DataFrame([]) 

for i in np.arange(0, 4): 
    if i % 2 == 0: 
     data = data.append(pd.DataFrame({'A': i, 'B': i + 1}, index=[0]), ignore_index=True) 
    else: 
     data = data.append(pd.DataFrame({'A': i}, index=[0]), ignore_index=True) 

print(data.head()) 

    A B 
0 0 1.0 
1 2 3.0 
2 3 NaN 
+0

ありがとう!私はそれを考えなかったばかげたばかだ。 – chizze

13

あなたが追加呼び出すたびに、パンダは、元のデータフレームに加え、新しい行のコピーを返す場所では発生しません。これは二次コピーと呼ばれ、O(N^2)操作であり、非常に遅くなります(特に、多くのデータがあるため)。

あなたの場合は、リストを追加して追加し、データフレームコンストラクタを呼び出すことをお勧めします。あなたはループせずにデータフレームを構築することができ

a_list = [] 
b_list = [] 
for data in my_data: 
    a, b = process_data(data) 
    a_list.append(a) 
    b_list.append(b) 
df = pd.DataFrame({'A': a_list, 'B': b_list}) 
del a_list, b_list 

タイミング

%%timeit 
data = pd.DataFrame([]) 
for i in np.arange(0, 10000): 
    if i % 2 == 0: 
     data = data.append(pd.DataFrame({'A': i, 'B': i + 1}, index=[0]), ignore_index=True) 
else: 
    data = data.append(pd.DataFrame({'A': i}, index=[0]), ignore_index=True) 
1 loops, best of 3: 6.8 s per loop 

%%timeit 
a_list = [] 
b_list = [] 
for i in np.arange(0, 10000): 
    if i % 2 == 0: 
     a_list.append(i) 
     b_list.append(i + 1) 
    else: 
     a_list.append(i) 
     b_list.append(None) 
data = pd.DataFrame({'A': a_list, 'B': b_list}) 
100 loops, best of 3: 8.54 ms per loop 
+0

さて、配列に保存してDataFrameを呼び出すのが実際には高速です。ありがとう! – chizze

2

n = 4 
data = pd.DataFrame({'A': np.arange(n)}) 
data['B'] = np.NaN 
data.loc[data['A'] % 2 == 0, 'B'] = data['A'] + 1 

の場合:

n = 10000 

これは少し速くなります。作品

%%timeit 
data = pd.DataFrame({'A': np.arange(n)}) 
data['B'] = np.NaN 
data.loc[data['A'] % 2 == 0, 'B'] = data['A'] + 1 

100 loops, best of 3: 3.3 ms per loop 

%%timeit 
a_list = [] 
b_list = [] 
for i in np.arange(n): 
    if i % 2 == 0: 
     a_list.append(i) 
     b_list.append(i + 1) 
    else: 
     a_list.append(i) 
     b_list.append(None) 
data1 = pd.DataFrame({'A': a_list, 'B': b_list}) 

100 loops, best of 3: 12.4 ms per loop