Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Simple stub of group_any
  • Loading branch information
WillAyd committed Feb 28, 2018
commit c5809cb9c27c21b6ff7d0381fee58464a349ba91
24 changes: 24 additions & 0 deletions pandas/_libs/groupby_helper.pxi.in
Original file line number Diff line number Diff line change
Expand Up @@ -791,3 +791,27 @@ def group_cummax_{{name}}(ndarray[{{dest_type2}}, ndim=2] out,
out[i, j] = mval

{{endfor}}

@cython.boundscheck(False)
@cython.wraparound(False)
def group_any(ndarray[uint8_t, ndim=2] out,
ndarray[int64_t] counts,
ndarray[:, :] values,
ndarray[int64_t] labels,
bint skipna):
cdef:
Py_ssize_t i, N
ndarray[uint8_t] mask

N, _ = (<object> labels).shape

out = np.zeros_like(out)
mask = values[:, 0].astype(np.bool)

for i in range(N):
lab = labels[i]
if lab < 0:
continue

if mask[lab]:
out[lab, 0] = 1