pandas巧用apply()和lambda方法

假设存在DataFrame存在两列数据col1和col2:

df = pd.DataFrame()

def function1(row):

    """ 公共函数 """

    print(row.col1)

    print(row.col2)

    return row.col1+row.col2

1. 直接对series使用apply()方法

def main1():

    """ 按照series使用apply"""

    df = pd.DataFrame()

    df['new_col'] = df['col'].apply(lambda row: function1(row))

2.直接对dataframe使用apply()方法

def main2():

    """ 按照dataframe使用apply"""

    df = df.apply(lambda row: function1(row), axis=1)    # 这里是对整行进行操作axis=1

备注: 使用这种方式处理数据, 比使用for循环效率要快很多, 并且function1()函数可以使用多种数据处理的方法.