Skip to content

Commit 060ae0c

Browse files
authored
Implement pandas export for 'select' queries (#107)
1 parent 4e18d5e commit 060ae0c

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

pydruid/query.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,10 @@ def export_pandas(self):
172172
nres = [list(v['event'].items()) + [('timestamp', v['timestamp'])]
173173
for v in self.result]
174174
nres = [dict(v) for v in nres]
175+
elif self.query_type == "select":
176+
nres = []
177+
for item in self.result:
178+
nres += [e.get('event') for e in item['result'].get('events')]
175179
else:
176180
raise NotImplementedError('Pandas export not implemented for query type: {0}'.format(self.query_type))
177181

0 commit comments

Comments
 (0)