Python Pandas Dataframe複製並重新排序列

Question

我有一個（可能）很容易的問題。我有以下DataFrame：

c1 | c2 | c3

a | b | 4
b | a | 1

現在，第3列僅依賴於第1列，第2列與第1列有相同的項目，但順序不同。我所要做的就是添加一個column4，其值與column3相同，但是按照column2的順序。所以結果如下所示：

c1| c2| c3|c4 

a | b | 4 | 1    
b | a | 1 | 4 

您有什麼想法嗎？

在此先感謝！

Answer 1

如果值是唯一的：

df['c4'] = df.set_index('c1').loc[df['c2'], 'c3'].values 
print (df) 
    c1 c2 c3 c4 
0 a b 4 1 
1 b a 1 4 

如果不是，它是有點complicted，因爲需要對cumcount創造獨特MultiIndex然後reindex：

print (df) 
    c1 c2 c3 
0 a b 5 
1 a a 4 
2 b a 1 

a = df.groupby('c1').cumcount() 
idx = pd.MultiIndex.from_arrays([df.groupby('c2').cumcount(), df['c2']]) 
print (idx) 
MultiIndex(levels=[[0, 1], ['a', 'b']], 
      labels=[[0, 0, 1], [1, 0, 0]], 
      names=[None, 'c2']) 

df['c4'] = df.set_index([a,'c1']).reindex(idx)['c3'].values 
print (df) 
    c1 c2 c3 c4 
0 a b 5 1 
1 a a 4 5 
2 b a 1 4 

Answer 2

不是像你一樣@jezrael答案但作品：

df['c4'] = df.apply(lambda x: dict(zip(df['c1'], df['c3'])).get(x['c2']), axis=1) 

    c1 c2 c3 c4 
0 a b 4 1 
1 b a 1 4