Всем привет. Появился вопрос по пандасу. например у меня есть датафрейм
col1 col2 col3
1 3 5
12 5 7
Задача такая. Нужно вытащить значения на определенных индексах для каждой колонки. например у меня есть массив [0,1,1]. по нему я должен вытащить нулевую строку для col1 и первую строку для col2 и col3
то есть аутпут должен быть
col1 col2 col3
1 5 7
Можно это решить пандосовскими функциями? Не циклами.
у меня решение такое:
import pandas as pd
import numpy as np
df = pd.DataFrame([[1,3,5,8],[12,5,7,12]], columns=['col1', 'col2', 'col3', 'col4'])
needed = [0, 1, 1, 0]
np.diag(df.iloc[needed])