Выбор значения из кадра данных с учетом строки и заголовка - PullRequest
0 голосов
/ 05 декабря 2018

У меня есть два фрейма данных:

ONE=pd.read_csv('ONE.csv')
value_one  value_two  
2          4          
3          1          
4          2          

TWO=pd.read_csv('TWO.csv')
X  1    2    3    4    5    6    7    8
1  a    c    j    a    d    c    c    d
2  c    k    a    d    c    c    d    e
3  f    c    k    a    d    c    c    d
4  c    k    a    d    c    c    d    j

Мне нужно создать дополнительный столбец в ОДНОМ фрейме данных (ONE['result']) в условиях:

, если значение value_one равно значению из заголовка фрейма данныхTWO

и value_two равно значению из ДВУХ кадров данных в столбце X,

установлено в новом общем столбце.

expected result:

value_one  value_two  result   
2          4          k
3          1          j
4          2          d

Я пытался: использовать для сравнения только заголовокif ONE[value_one]==TWO.iloc[0]

Спасибо, С.

1 Ответ

0 голосов
/ 05 декабря 2018

lookup

Вы можете lookup свой второй кадр данных:

df_two = df_two.set_index('X')  # set 'X' column as index
df_two.columns = df_two.columns.astype(int)  # ensure column labels are numeric

df_one['result'] = df_two.lookup(df_one['value_two'], df_one['value_one'])

print(df_one)

   value_one  value_two result
0          2          4      k
1          3          1      j
2          4          2      d
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...