小编pie*_*oic的帖子

将函数应用/组合 N 次到 Pandas 列,每行 N 不同

假设我们有这个简单的 pandas.DataFrame:

import pandas as pd

df = pd.DataFrame(
  columns=['quantity', 'value'],
  data=[[1, 12.5], [3, 18.0]]
)

>>> print(df)
   quantity  value
0         1   12.5
1         3   18.0
Run Code Online (Sandbox Code Playgroud)

我想创建一个新列,例如modified_value,将函数应用于该value列N 次,N 是该quantity列。假设该函数为new_value = round(value/2, 1),则预期结果为:

   quantity  value  modified_value
0         1   12.5            6.2   # applied 1 time
1         3   9.0             1.1   # applied 3 times, 9.0 -> 4.5 -> 2.2 -> 1.1
Run Code Online (Sandbox Code Playgroud)

这样做的优雅/矢量化方式是什么?

python pandas

5
推荐指数
1
解决办法
256
查看次数

标签 统计

pandas ×1

python ×1