python - 在 SFrame 中选择特定行

Question

我对如何在 SFrame 数组中选择特定行感到困惑。我可以在这里选择第一行：

sf
+-------------------------------+
|               X1              |
+-------------------------------+
| [0.0, 0.0, 0.0, 0.0, 0.0, ... |
[100 rows x 1 columns]

sf[:1]
+-------------------------------+
|               X1              |
+-------------------------------+
| [0.0, 0.0, 0.0, 0.0, 0.0, ... |
[1 rows x 1 columns]

sf[:2]
+-------------------------------+
|               X1              |
+-------------------------------+
| [0.0, 0.0, 0.0, 0.0, 0.0, ... |
| [0.0, 0.0, 0.0, 0.0, 0.0, ... |
[2 rows x 1 columns]

type(sf[:1])
graphlab.data_structures.sframe.SFrame

在这里，我试图只获得第 2 行

sf[:,2]
# TypeError: Invalid key type: must be str, bytes or type

如何选择数据框中的任何行？

Answer 1

您可以选择一行：

import graphlab as gl
sf = gl.SFrame({'a':[1,2,3], 'b':[2,9,1]})
# select first row
print sf[0]
# select second row
print sf[1] 
# and so on
# convert first row to an SFrame
sf_one_raw = sf[0:1]
# convert second row to an SFrame
sf_one_raw = sf[1:2]