pandas DataFrame 中映射列的最有效方法

Question

我想知道当列中需要简单映射时， map方法是否是最佳选择，因为使用map或apply 通常是个坏主意。

Answer 1

我比较了以下简单案例的以下功能。 如果您有更好的选择，请分享。

# Case - Map the random number to its string
import pandas as pd
import numpy as np
df = pd.DataFrame(np.random.randint(1,7,size=(5000,1)), columns=['A'])
dikt = {1:'1',2:'2',3:'3',4:'4',5:'5',6:'6'}

首先 function - 使用map方法：

def f1():
    df1 = df.copy()
    df1['B'] = df['A'].map(dikt)
    return df1

结果：

第二个 function - 在列中使用to_list方法：

def f2():
    df2 = df.copy()
    column_list = df2['A'].tolist()
    df2['B'] = [dikt[i] for i in column_list]
    return df2

结果：