Python:创建新列并从其他行复制值，这是当前行的交换

Question

我有一个dataframe，它有3列：

import pandas as pd

d = {'A':  ['left', 'right', 'east', 'west', 'south', 'north'], 'B': ['right', 'left', 'west', 'east', 'north', 'south'], 'VALUE': [0, 1, 2, 3, 4, 5]}

df = pd.DataFrame(d)

Dataframe看起来如下：

  A       B   VALUE
left    right   0
right   left    1
east    west    2
west    east    3
south   north   4
north   south   5

我正在尝试创建一个新的列VALUE_2，它应该包含来自同一个Dataframe中交换行的值。

例:右左值为0，左-右值为1，我希望新列中的交换值如下：

  A       B   VALUE VALUE_2
left    right   0     1
right   left    1     0
east    west    2     3
west    east    3     2
south   north   4     5
north   south   5     4

我试过：

for row_num, record in df.iterrows():
    A = df['A'][index]
    B = df['B'][index]
    if(pd.Series([record['A'] == B, record['B'] == A).all()):
        df['VALUE_2'] = df['VALUE']

我很惊讶，投入将是非常感谢的。

Answer 1

map在Series中的应用

df['VALUE_2'] = df['A'].map(df.set_index('B')['VALUE'])
print (df)
       A      B  VALUE  VALUE_2
0   left  right      0        1
1  right   left      1        0
2   east   west      2        3
3   west   east      3        2
4  south  north      4        5
5  north  south      5        4

Answer 2

只是一个更详细的回答：

import pandas as pd

d = {'A':  ['left', 'right', 'east', 'west', 'south', 'north'], 'B': ['right', 'left', 'west', 'east', 'north', 'south'], 'VALUE': [0, 1, 2, 3, 4, 5]}

df = pd.DataFrame(d)

pdf = pd.DataFrame([])
for idx, item in df.iterrows():
    indx = list(df['B']).index(str(df['A'][idx]))
    pdf = pdf.append(pd.DataFrame({'VALUE_2': df.iloc[indx][2]}, index=[0]), ignore_index=True)

print(pdf)
data = pd.concat([df, pdf], axis=1)
print(data)