2

我发现自己经常编写这种模式：

tmp = <some operation>
result = tmp[<boolean expression>]
del tmp

... where<boolean expression>应理解为涉及 tmp. 的布尔表达式。（目前，tmp它始终是一个 pandas 数据框，但我想如果我使用 numpy ndarrays 会出现相同的模式——不确定。）

例如：

tmp = df.xs('A')['II'] - df.xs('B')['II']
result = tmp[tmp < 0]
del tmp

从del tmp最后可以猜到，创建的唯一原因tmp是我可以在应用于它的索引表达式中使用涉及它的布尔表达式。

我很想消除对这种（否则无用的）中间体的需求，但我不知道有任何有效的方法来做到这一点。（如果我错了，请纠正我！）

作为第二好的，我想将这种模式推向一些辅助功能。问题是找到一种体面的方式来传递<boolean expression>给它。我只能想到不雅的。例如：

def filterobj(obj, criterion):
    return obj[eval(criterion % 'obj')]

这实际上有效2：

filterobj(df.xs('A')['II'] - df.xs('B')['II'], '%s < 0')

# Int
# 0     -1.650107
# 2     -0.718555
# 3     -1.725498
# 4     -0.306617
# Name: II

...但是使用eval总是让我觉得所有的东西都令人讨厌...如果有其他方法，请告诉我。

1例如，我能想到的任何涉及filter内置的方法都可能效率低下，因为它会通过“在 Python 中”在熊猫（或 numpy）对象上迭代来应用标准（一些 lambda 函数）......

2上面最后一个表达式中使用的定义是df这样的：

import itertools
import pandas as pd
import numpy as np
a = ('A', 'B')
i = range(5)
ix = pd.MultiIndex.from_tuples(list(itertools.product(a, i)),
                               names=('Alpha', 'Int'))
c = ('I', 'II', 'III')
df = pd.DataFrame(np.random.randn(len(idx), len(c)), index=ix, columns=c)
4

2 回答 2

1

这是尽可能简洁的：

(df.xs('A')['II'] - df.xs('B')['II']).apply(lambda x: x if (x<0) else np.nan).dropna()

Int
0     -4.488312
1     -0.666710
2     -1.995535
Name: II
于 2013-02-20T19:07:24.710 回答
1

由于 Python 的工作方式，我认为这会很困难。我只能想到只能让你部分获得成功的技巧。就像是

def filterobj(obj, fn):
    return obj[fn(obj)]

filterobj(df.xs('A')['II'] - df.xs('B')['II'], lambda x: x < 0)

应该工作，除非我错过了什么。以这种方式使用 lambda 是延迟评估的常用技巧之一。

大声思考：可以制作一个this不被评估但只是作为一个表达式存在的对象，比如

>>> this
this
>>> this < 3
this < 3
>>> df[this < 3]
Traceback (most recent call last):
  File "<ipython-input-34-d5f1e0baecf9>", line 1, in <module>
    df[this < 3]
[...]
KeyError: u'no item named this < 3'

然后要么将特殊情况处理this成熊猫，要么仍然具有类似的功能

def filterobj(obj, criterion):
    return obj[eval(str(criterion.subs({"this": "obj"})))]

（有足够的工作，我们可能会失去eval，这只是概念证明）之后类似

>>> tmp = df["I"] + df["II"]
>>> tmp[tmp < 0]
Alpha  Int
A      4     -0.464487
B      3     -1.352535
       4     -1.678836
Dtype: float64
>>> filterobj(df["I"] + df["II"], this < 0)
Alpha  Int
A      4     -0.464487
B      3     -1.352535
       4     -1.678836
Dtype: float64

会工作。我不确定这是否值得头疼，不过，Python 根本不适合这种风格。

于 2013-02-20T19:32:49.540 回答