FilesExpand file tree

 master

/

pandas_lib.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

27 lines (20 loc) · 738 Bytes

 master

/

pandas_lib.py

Copy path

Top

File metadata and controls

• Code
• Blame

27 lines (20 loc) · 738 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

from pandas import DataFrame

from pythonql.PQTuple import PQTuple

def to_df(query_res,columns=[]):

it = iter(query_res)

try:

first = next(it)

if isinstance(first,PQTuple) and not columns:

schema = first.schema.items()

cols = [k for (k,v) in sorted(schema, key=lambda x:x[0])]

return DataFrame.from_records([first] + [x for x in it], columns=cols)

elif columns:

return DataFrame.from_records([first] + [x for x in it], columns=columns)

else:

return DataFrame([first] + [x for x in it])

except:

return DataFrame.from_records([],columns=columns)

def wrap_df(df):

schema = { n:i for (i,n) in enumerate(df.columns) }

for t in df.itertuples(False):

yield PQTuple(t,schema)