python - 在 Pandas 中用自身的功能替换列？

Question

我目前迷失在 pandas 文档的深处。我的问题是:

我有一个简单的数据框

col1  col2
 1     A
 4     B 
 5     X   

我的目标是应用类似的东西:

 df['col1'] = df['col1'].apply(square)

其中 square 是一个明确定义的函数。但是此操作会引发错误警告(并产生不正确的结果)

试图在 DataFrame 的切片副本上设置一个值。尝试使用 .loc[row_indexer,col_indexer] = value 代替

我无法理解这一点，也无法理解它指向的文档。我的工作流程是线性的(如果这使得更广泛的解决方案可行)。

Pandas 0.17.1 和 Python 2.7

非常感谢所有帮助。

Answer 1

它适合我( Pandas 0.18.1):

In [31]: def square(x):
   ....:     return x ** 2
   ....:

In [33]: df
Out[33]:
   col1 col2
0     1    A
1     4    B
2     5    X

In [35]: df.col1 = df.col1.apply(square)

In [36]: df
Out[36]:
   col1 col2
0     1    A
1    16    B
2    25    X

PS 这也可能取决于您的函数的实现...