在hy中使用pandas loc

Question

我想在hy中执行以下操作：

from StringIO import StringIO
import pandas as pd

s = """sepal_length  sepal_width  petal_length  petal_width species
0           5.1          3.5           1.4          0.2  setosa
1           4.9          3.0           1.4          0.2  setosa
2           4.7          3.2           1.3          0.2  setosa
3           4.6          3.1           1.5          0.2  setosa
4           5.0          3.6           1.4          0.2  setosa
5           5.4          3.9           1.7          0.4  setosa
6           4.6          3.4           1.4          0.3  setosa
7           5.0          3.4           1.5          0.2  setosa
8           4.4          2.9           1.4          0.2  setosa
9           4.9          3.1           1.5          0.1  setosa"""

df = pd.read_table(StringIO(s), sep="\s+")

df.loc[df.sepal_length > 4.5]

我如何做最后一句话？

我试过了(.loc df (> df.sepal_length 4.5))

但它只返回一个locindexer。

Answer 1

有两种方法可以做到这一点：

1. 使用. macro：

   (. df loc [(> df.sepal-length 4.5)])
   

2. 使用get：

   (get df.loc (> df.sepal-length 4.5))
   

Protip：总是尝试在你的Hy文件上运行hy2py。它显示了生成的Python的样子。输出并不总是有效的语法，但它会显示编译成什么的内容。这两个都被编译为df.loc[(df.sepal_length > 4.5)]。

还有一件事：注意我使用了sepal-length。 Hy将标识符中的破折号转换为下划线，这与sepal_length相同，但它被认为是更好的风格。