在所有 Dataframe 行中应用函数的返回值

Question

我正在尝试从新列中 Dataframe 的所有行中的函数分配一个值。 即下面的 df 应包含称为random number新列，其中包含在get_random()函数中生成的随机数。

import pandas as pd
from random import random

def get_random():
    return random()

d = {'col1': [1, 2], 'col2': [3, 4]}
df = pd.DataFrame(data=d)

我尝试了下面的方法，但出现错误get_random takes 0 positional arguments

df['random number'] = df.apply(get_random)

Answer 1

你可以这样做

df['random number'] = df.apply(lambda x: random())

或将 get_random 更改为

def get_random(x):
    return random()

Answer 2

只需更改函数random ：

def get_random(x):
    return random()

因为 apply 期望函数接受输入。

Answer 3

您可以使用列表理解

df['random number'] = [get_random() for i in range(len(df))]