Как составить логику выбора данных с DataFrame?

Question

[beckmiller]

Всем привет!
У меня есть данные

DATA	X	Y
Train	1	0
Train	0	1
Test		1      1
Train	1	0
Test		1      1

Как мне для всех значений Test взять значении X и Y и записать в csv файл в pandas.
Чтобы результат был такой:

X       Y
1       1
1       1

Answer 1

[Gomonov]

df[df['DATA'] == 'Test'].drop(['DATA'], axis=1).to_csv('data.csv', mode='w', index=False)

Comments

[beckmiller]

Спасибо