Выберите несколько столбцов на основе ГДЕ в кадре данных

Question

Итак, я работаю с Blaze и хотел выполнить этот запрос на фрейме данных:

SELECT col1,col2 FROM table WHERE col1 > 0

За SELECT *, это работает: d[d.col1 > 0], Но я хочу col1 а также col2 только вместо всех столбцов. Как я должен идти об этом?

Заранее спасибо!

Редактировать: здесь я создаю d как: d = Data('postgresql://uri')

Answer 1

Это также работает: d[d.col1 > 0][['col1','col2']]

Answer 2

Я думаю, что вы можете использовать сначала подмножество, а затем логическое индексирование:

 print (d)
   col1  col2  col3
0    -1     4     7
1     2     5     8
2     3     6     9

d = d[['col1','col2']]
print (d)
   col1  col2
0    -1     4
1     2     5
2     3     6

print (d[d.col1 > 0])
   col1  col2
1     2     5
2     3     6

Это так же, как:

print (d[['col1','col2']][d.col1 > 0])
   col1  col2
1     2     5
2     3     6