Выберите значение столбца Dataframe Pandas, основанное на другом значении столбца

Question

Потянув мои волосы за это, хотя я уверен, что кто-то предоставит простой ответ.

data = [{'check': None, 'iterator': 1, 'x1': 1, 'x2': 2, 'x3':3}, 
     {'check': None, 'iterator': 2, 'x1': 1, 'x2': 2, 'x3':3}, 
     {'check': None, 'iterator': 3, 'x1': 1, 'x2': 2 , 'x3':3}] 
df = pd.DataFrame(data) 
display(df) 

Я пытаюсь заполнить столбец проверки путем сдвига ИНГ вправо на значение в колонке «итератор». То есть первая строка «проверки» будет 1, вторая будет равна 2, третья - 3.

Это упрощение гораздо большего набора данных, над которым я работаю, поэтому я бы оценил векторизованное код.

Answer 1

Вы можете использовать numpy «s advanced indexing:

df['check'] = df.filter(like="x").values[np.arange(len(df)), df.iterator - 1] 

---

Чтобы избежать двусмысленности, вот другой пример:

data = [{'check': None, 'iterator': 2, 'x1': 3, 'x2': 4, 'x3':3}, 
     {'check': None, 'iterator': 1, 'x1': 1, 'x2': 5, 'x3':3}, 
     {'check': None, 'iterator': 2, 'x1': 2, 'x2': 2 , 'x3':1}] 
df = pd.DataFrame(data) 

df['check'] = df.filter(like="x").values[np.arange(len(df)), df.iterator-1] 
df