Skip to content

Commit 3ff4c85

Browse files
authored
Fix KeyError in get_group (#109)
1 parent f79af86 commit 3ff4c85

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

lib/charty/table_adapters/pandas_adapter.rb

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,12 @@ def [](key)
208208
when PyCall::Tuple
209209
key
210210
else
211-
PyCall::Tuple.new(*Array(key))
211+
key = Array(key)
212+
if key.length == 1 && key[0].is_a?(Integer)
213+
key[0]
214+
else
215+
PyCall::Tuple.new(*key)
216+
end
212217
end
213218
Charty::Table.new(@groupby.get_group(key))
214219
end

0 commit comments

Comments
 (0)