H2OFrame列到数组:最快的方法

Question

假设我有一个名为df的H2OFrame。从所述帧中获得列x的值作为numpy数组的最快方法是什么？

一个人可以做

x_array = df['x'].as_data_frame()['x'].values

但这似乎不必要地冗长。特别是通过具有as_data_frame的pandas DataFrame似乎是多余的。我希望有更优雅的东西，比如df['x'].to_array()。但我找不到。

Answer 1

这里是另一种方式。然而，我不确定它是否更快。我使用h2o.as_list((函数将列转换为列表，然后使用np.array((函数把列表转换为数组。

import h2o
import numpy as np
h2o.init()
# Using sample dataset from H2O
train = h2o.import_file("https://s3.amazonaws.com/erin-data/higgs/higgs_train_10k.csv")

## Creating np array from h2o frame column  
np.array(h2o.as_list(train['x1']))