python - 将函数应用于数据框中的列 - python

Question

我编写了一个函数来消除数据框中值的空格。

def eliminate_space(words):
    return words.replace(' ', '')

现在我想将该函数应用于数据帧“单词”列中的每一行。

df['words']=df['words'].apply(eliminate_space(words))

它表示名称“words”未定义。如何调整我的代码以便能够将该函数应用于数据帧该列中的每个值？

Answer 1

您可以将 apply 与 lambda 一起用于 row 的该列:

df['words']=df['words'].apply(lambda row: eliminate_space(row))

使用示例进行测试:

df = pd.DataFrame({'words':['word is 1', 'word is 2', 'word is 3']})
print(df)

初始数据框:

     words
0  word is 1
1  word is 2
2  word is 3

使用函数:

df['words']=df['words'].apply(lambda row: eliminate_space(row))

结果:

     words
0  wordis1
1  wordis2
2  wordis3