2019年2月17日日曜日

Python3 cuDF

https://github.com/rapidsai/cuml/issues/26

import cudf

# convert numpy array to cuDF dataframe
def np2cudf(df):
    df = pd.DataFrame({'fea%d'%i:df[:,i] for i in range(df.shape[1])})
    pdf = cudf.DataFrame()
    for c,column in enumerate(df):
      pdf[str(c)] = df[column]
    return pdf

img_cudf = np2cudf(img_array)
print("input:")
print(img_cudf)


0 件のコメント:

コメントを投稿