Выбор, заказ и переименование столбцов в пандах

Question

Я пытаюсь найти R's select эквивалент функции в пандах. Существует ссылка для основ, но она не дает указания, что я хочу сделать!

raw_data = {'patient': [1, 1, 1, 2, 2],
        'obs': [1, 2, 3, 1, 2],
        'treatment': [0, 1, 0, 1, 0],
        'score': ['strong', 'weak', 'normal', 'weak', 'strong']}
df = pd.DataFrame(raw_data, columns = ['patient', 'obs', 'treatment', 'score'])


df.rename(columns = {'treatment':'treat'},inplace=True)


df = df.loc[:, ['treat','score','obs']]

Out[89]: 
   treat   score  obs
0      0  strong    1
1      1    weak    2
2      0  normal    3
3      1    weak    1
4      0  strong    2

Мы можем сделать это с R's dplyr

select(df, treat=treatment, score, obs)  that's it. 

Как я могу сделать это с пандами всего одной строкой кода, выбирая, упорядочивая и переименовывая?

Answer 1

В пандах нет эквивалента для выбора и переименования в одном методе, если необходимо использовать аналогично вашему решению:

df = df.rename(columns = {'treatment':'treat'})[['treat','score','obs']]
#alternative
#df = df[['treatment','score','obs']].rename(columns = {'treatment':'treat'})
print (df)
   treat   score  obs
0      0  strong    1
1      1    weak    2
2      0  normal    3
3      1    weak    1
4      0  strong    2