更简洁的过滤列表的方法？

Question

如果我有清单

a=[4, 5, 6]

据我所知，最简单的过滤方法是：

[i for i in a if a<6]

现在我刚刚被介绍到数据帧，其中的数据帧像

df = pd.DataFrame({'a':[4, 5, 6], 'b':[7, 1, 2]})

我可以通过指定元素和条件来应用（行）过滤器：

df[df['a']<6]

这似乎比过滤列表的方式更简洁，而且可能更不容易混淆（一旦你习惯了）。无法通过简单地在 [] 中指定条件来应用列表过滤器，例如

a[<6]

很明显，目前还没有这样实现，但是现在的方法是不是比较冗长？为什么不能简化？

Answer 1

你有上面最简单的方法。但是，您可以使用以下

filtered_list = filter(lambda k: k < 6, original_list)

这看起来很棒，但我仍然对列表理解情有独钟。

Answer 2

您不能完全拥有您要求的语法，但如果您想创建自己的列表类，您可以拥有一个同样简洁的：

class List(list):
    def __lt__(self, other):
        return List(i for i in self if i < other)

a = List([4,5,6])
b = a < 6
assert b == [4,5]

Answer 3

是的，可以构建 Python 语言，以便 a[<6] 进行您想要的过滤，但是每个 Python 程序员和每个 Python 编译器都必须学习这种特殊语法，以便在一些特殊情况下节省一点源代码：>、>=、==、

Pandas 会像您展示的那样做一些事情，但它是为大量数值分析而构建的，因此语法糖可能更值得付出代价。此外，pandas 倾向于提供大量受 R 语言启发的语法糖，因此它不是非常惯用的 python。