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

結果：