如何在熊猫数据框中使用应用时创建列时间戳？

Question

我将一些函数应用于熊猫数据框列：

def foo(x):
     return 1 + x

然后，我将函数应用于列：

df['foo'] = df['a_col'].apply(foo)

如何返回一个列，其中包含函数 foo 完成所需的毫秒数？例如：

A time_milisecs
2 0.1
4 0.2
4 0.3
3 0.3
4 0.2

其中A 是包含求和结果的列。

Answer 1

您可以使用time 模块。假设您还希望通过计算创建一个新系列，您可以输出一个元组序列，然后转换为数据帧并分配回两个系列。

这是一个演示：

import time

df = pd.DataFrame({'A': [2, 4, 4, 3, 4]})

def foo(x):
    tstart = time.time()
    time.sleep(0.25)
    tend = time.time()
    return 1 + x, (tend-tstart) * 10**3

df[['B', 'B_time']] = pd.DataFrame(df['A'].apply(foo).values.tolist())

print(df)

   A  B      B_time
0  2  3  250.014544
1  4  5  250.014305
2  4  5  250.014305
3  3  4  250.014305
4  4  5  250.014067

在 Python 3.7 中，您可以使用 time.process_time_ns，它以纳秒为单位测量时间。