New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Show group values when printing grouped dataframe #1632
Changes from all commits
1b2798d
ff0b463
84ff9ba
2762f10
461d345
8a6301d
3978895
43e8378
085e672
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -6,31 +6,51 @@ function Base.show(io::IO, gd::GroupedDataFrame; | |
rowlabel::Symbol = :Row, | ||
summary::Bool = true) | ||
N = length(gd) | ||
keys = join(':' .* string.(names(gd.parent)[gd.cols]), ", ") | ||
keynames = names(gd.parent)[gd.cols] | ||
parent_names = names(gd.parent) | ||
keys = join(':' .* string.(keynames), ", ") | ||
|
||
keystr = length(gd.cols) > 1 ? "keys" : "key" | ||
groupstr = N > 1 ? "groups" : "group" | ||
summary && print(io, "$(typeof(gd)) with $N $groupstr based on $keystr: $keys") | ||
if allgroups | ||
for i = 1:N | ||
nrows = size(gd[i], 1) | ||
rows = nrows > 1 ? "rows" : "row" | ||
print(io, "\nGroup $i: $nrows $rows") | ||
|
||
identified_groups = [':' * string(parent_names[col], " = ", first(gd[i][col])) | ||
for col in gd.cols] | ||
|
||
print(io, "\nGroup $i ($nrows $rows): ") | ||
join(io, identified_groups, ", ") | ||
|
||
show(io, gd[i], summary=false, | ||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Maybe we shouldn't print the grouping columns, since they're listed above? There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I think it reads well. There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. As you like. But just to make sure we're talking about the same thing: I was suggesting we could drop the grouping columns from the data frame we print below, since all values are equal within a given group. |
||
allrows=allrows, allcols=allcols, rowlabel=rowlabel) | ||
end | ||
else | ||
if N > 0 | ||
nrows = size(gd[1], 1) | ||
rows = nrows > 1 ? "rows" : "row" | ||
print(io, "\nFirst Group: $nrows $rows") | ||
|
||
identified_groups = [':' * string(parent_names[col], " = ", first(gd[1][col])) | ||
for col in gd.cols] | ||
|
||
print(io, "\nFirst Group ($nrows $rows): ") | ||
join(io, identified_groups, ", ") | ||
|
||
show(io, gd[1], summary=false, | ||
allrows=allrows, allcols=allcols, rowlabel=rowlabel) | ||
end | ||
if N > 1 | ||
print(io, "\n⋮\n") | ||
nrows = size(gd[N], 1) | ||
rows = nrows > 1 ? "rows" : "row" | ||
print(io, "Last Group: $nrows $rows") | ||
|
||
identified_groups = [':' * string(parent_names[col], " = ", first(gd[N][col])) | ||
for col in gd.cols] | ||
print(io, "\n⋮") | ||
print(io, "\nLast Group ($nrows $rows): ") | ||
join(io, identified_groups, ", ") | ||
|
||
show(io, gd[N], summary=false, | ||
allrows=allrows, allcols=allcols, rowlabel=rowlabel) | ||
end | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I've just realized that when the grouping column is a string or symbol, it's printed as e.g.
:x = a
rather than:x = "a"
or:x = :a
. I guess we should change this (usingrepr
orourshowcompact
)?There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It should be
repr
. I will make a quick PR.There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
#1684