pandas - 是否有相当于 df.irow 的 graphlab？

Question

我需要按索引在我的 sframe 中挑选几行。是否有与 pandas df.irow() 等效的 graphlab 命令？

Answer 1

graphlab 中没有与DataFrame.iloc（以前irow）的直接等价物。实现相同目的的一种方法是添加一列行号并使用该filter_by方法。假设我只想获得第一行和第三行：

import graphlab
sf = graphlab.SFrame({'x': ['a', 'b', 'a', 'c']})
sf = sf.add_row_number('row_id')
new_sf = sf.filter_by(values=[0, 2], column_name='row_id')