熊猫 - '系列'对象没有属性' colNames'何时使用apply()

时间:2016-11-09 11:53:48

标签: python pandas

我需要使用lambda函数来逐行计算。例如,创建一些数据帧

import pandas as pd
import numpy as np

def myfunc(x, y):
    return x + y

colNames = ['A', 'B']
data = np.array([np.arange(10)]*2).T

df = pd.DataFrame(data, index=[range(0, 10)], columns=colNames)

使用' myfunc'这确实有效

df['D'] = (df.apply(lambda x: myfunc(x.A, x.B), axis=1))

但是第二种情况不起作用!

df['D'] = (df.apply(lambda x: myfunc(x.colNames[0], x.colNames[1]), axis=1))

给出错误

AttributeError: ("'Series' object has no attribute 'colNames'", u'occurred at index 0')

我真的需要使用第二种情况(使用列表访问colNames),这会产生错误,有关如何执行此操作的任何线索?

感谢

1 个答案:

答案 0 :(得分:8)

使用df.apply()时,DataFrame的每一行都将作为pandas系列传递给lambda函数。然后,框架的列将成为系列的索引,您可以使用series[label]访问值。

所以这应该有效:

df['D'] = (df.apply(lambda x: myfunc(x[colNames[0]], x[colNames[1]]), axis=1))