python - 如何在 Pandas 数据框中 self 引用列？

Question

在 Python 的 Pandas 中，我这样使用数据框:

drinks = pandas.read_csv(data_url)

其中 data_url 是指向 CSV 文件的字符串 URL

当为所有“轻度饮酒者”索引框架时，其中轻度饮酒者由 1 杯饮料构成，如下所示:

drinks.light_drinker[drinks.light_drinker == 1]

有没有更像 DRY 的方式来 self 引用“ parent ”？ IE。像这样的东西:

drinks.light_drinker[self == 1]

Answer 1

您现在可以使用 query或 assign取决于您的需要:

drinks.query('light_drinker == 1')

或改变 df:

df.assign(strong_drinker = lambda x: x.light_drinker + 100)

旧答案

目前还没有，但是正在讨论对您的想法进行改进here .对于简单的情况，where 可能就足够了。新的 API 可能如下所示:

df.set(new_column=lambda self: self.light_drinker*2)