有没有相当于df.irow的graphlab？

Question

我需要通过索引在我的sframe中挑选几行。 pandas df.irow（）是否有等效的graphlab命令？

Answer 1

graphlab中没有与DataFrame.iloc（之前为irow）的直接对等关系。实现相同目标的一种方法是添加一列行号并使用filter_by方法。假设我只想获得第1行和第3行：

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')