join
.from_delayed メソッドによって生成された dask データフレームに適用すると、予期しない結果が得られました。これを次の例で示したいと思います。これは 3 つの部分で構成されています。
- メソッドを介してdaskデータフレームを生成し、それを介して
from_delayed
生成されたdaskデータフレームと結合しますfrom_pandas
- メソッドを使用して、両方のデータフレームを pandas データフレームに変換します
compute
。(1)のように参加した from_delayed
メソッドを介して生成された dask データフレームを、 を使用して pandas に変換しcompute
ます。Aferwards は、 を使用してそれを dask に戻しfrom_pandas
ます。次に、(1) のように結合します。
次のコードを検討してください。
import dask.dataframe
import pandas as pd
# functions for generating a dask dataframe
def get_pdf(character):
'''constructs a pandas dataframe with indexes [character]1, ..., [character]5'''
index = [character + str(i) for i in range(5)]
return pd.DataFrame({'A':[1,2,3,4,5]}, index = index)
def get_ddf():
'''constructs dask dataframe out of pandas dataframes via the .from-delayed method with indexes A1, A2, A3, ... F3, F3, F4'''
delayed_list = [dask.delayed(get_pdf)(x) for x in 'ABCDEF']
return dask.dataframe.from_delayed(delayed_list)
#generate dask dataframes, that will be joined
ddf1 = get_ddf()
ddf2 = dask.dataframe.from_pandas(pd.DataFrame({'B': [1,2,3]}, index = ['A0', 'B1', 'C3']), npartitions = 2)
#recreate ddf1 by converting it to a pandas dataframe and afterwards to a dask dataframe
ddf1_from_pandas = dask.dataframe.from_pandas(ddf1.compute(), npartitions = 3)
#compute joins
dask_from_delayed_join = ddf1.join(ddf2, how = 'inner')
pandas_join = ddf1.compute().join(ddf2.compute(), how = 'inner')
dask_from_pandas_join = ddf1_from_pandas.join(ddf2, how = 'inner')
dask_from_delayed_join
3 つの結果 ( 、pandas_join
、 )はすべて同じになると思いdask_from_pandas_join
ます。
ただし、最初の結果は他の結果とは異なります。
print(dask_from_delayed_join.compute())
:
Empty DataFrame
Columns: [A, B]
Index: []
print(pandas_join)
:
A B
A0 1 1
B1 2 2
C3 4 3
print(dask_from_pandas_join.compute())
:
A B
A0 1 1
B1 2 2
C3 4 3
何が起こっている?