Skip to content

ENH: Add support for complex weights in np.bincount #23641

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.

Sign up for GitHub

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

Open
wants to merge 5 commits into
base: main
Choose a base branch
from
Open

ENH: Add support for complex weights in np.bincount #23641

Changes from 1 commit
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
03003a4
Add support for complex weights
alex-gregory-ds Apr 22, 2023
66ae571
Add support for integers
alex-gregory-ds Apr 28, 2023
c9ef986
Switch to unsigned integers for the weights
alex-gregory-ds Apr 28, 2023
2cfe3a7
Change to npy_uintp to reflect type at initialization
alex-gregory-ds Apr 28, 2023
6ee969c
Initialise weights for the complex case
alex-gregory-ds Apr 28, 2023
File filter

Filter by extension

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
Change to npy_uintp to reflect type at initialization
  • Loading branch information
@alex-gregory-ds
alex-gregory-ds committed Apr 28, 2023
commit 2cfe3a7307929389ee313a4232f0a2d95bc1b4d9
4 changes: 2 additions & 2 deletions numpy/core/src/multiarray/compiled_base.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -184,11 +184,11 @@ arr_bincount(PyObject *NPY_UNUSED(self), PyObject *const *args,
}
}
if (weight == Py_None) {
ans = (PyArrayObject *)PyArray_ZEROS(1, &ans_size, NPY_INTP, 0);
ans = (PyArrayObject *)PyArray_ZEROS(1, &ans_size, NPY_UINTP, 0);
if (ans == NULL) {
goto fail;
}
ians = (npy_intp *)PyArray_DATA(ans);
ians = (npy_uintp *)PyArray_DATA(ans);
NPY_BEGIN_ALLOW_THREADS;
for (i = 0; i < len; i++)
ians[numbers[i]] += 1;
Expand Down
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy