(group_ids, th)
| 166 | |
| 167 | |
| 168 | def remove_small_group(group_ids, th): |
| 169 | unique_elements, counts = np.unique(group_ids, return_counts=True) |
| 170 | result = group_ids.copy() |
| 171 | for i, count in enumerate(counts): |
| 172 | if count < th: |
| 173 | result[group_ids == unique_elements[i]] = -1 |
| 174 | |
| 175 | return result |
| 176 | |
| 177 | |
| 178 | def pairwise_indices(length): |
no outgoing calls
no test coverage detected