Skip to content

firest_true_* performance study #2

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 1 commit into
base: main
Choose a base branch
from
Open

firest_true_* performance study #2

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
83 changes: 61 additions & 22 deletions doc/articles/first_true_2d.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,9 @@
from itertools import repeat

from arraykit import first_true_2d as ak_first_true_2d
from arrayredox import first_true_2d as ar_first_true_2d
from arrayredox import first_true_2d as ar_first_true_2d_a
from arrayredox import first_true_2d_b as ar_first_true_2d_b

import arraykit as ak

import matplotlib.pyplot as plt
Expand Down Expand Up @@ -58,41 +60,72 @@ def __call__(self):


#-------------------------------------------------------------------------------
class ARFirstTrueAxis0Forward(ArrayProcessor):
NAME = 'ar.first_true_2d(forward=True, axis=0)'
class ARFirstTrueAAxis0Forward(ArrayProcessor):
NAME = 'ar.first_true_2d_a(forward=True, axis=0)'
SORT = 10

def __call__(self):
_ = ar_first_true_2d(self.array, forward=True, axis=0)
_ = ar_first_true_2d_a(self.array, forward=True, axis=0)

class ARFirstTrueAxis1Forward(ArrayProcessor):
NAME = 'ar.first_true_2d(forward=True, axis=1)'
class ARFirstTrueAAxis1Forward(ArrayProcessor):
NAME = 'ar.first_true_2d_a(forward=True, axis=1)'
SORT = 11

def __call__(self):
_ = ar_first_true_2d(self.array, forward=True, axis=1)
_ = ar_first_true_2d_a(self.array, forward=True, axis=1)

class ARFirstTrueAxis0Reverse(ArrayProcessor):
NAME = 'ar.first_true_2d(forward=False, axis=0)'
class ARFirstTrueAAxis0Reverse(ArrayProcessor):
NAME = 'ar.first_true_2d_a(forward=False, axis=0)'
SORT = 12

def __call__(self):
_ = ar_first_true_2d(self.array, forward=False, axis=0)
_ = ar_first_true_2d_a(self.array, forward=False, axis=0)

class ARFirstTrueAxis1Reverse(ArrayProcessor):
NAME = 'ar.first_true_2d(forward=False, axis=1)'
class ARFirstTrueAAxis1Reverse(ArrayProcessor):
NAME = 'ar.first_true_2d_a(forward=False, axis=1)'
SORT = 13

def __call__(self):
_ = ar_first_true_2d(self.array, forward=False, axis=1)
_ = ar_first_true_2d_a(self.array, forward=False, axis=1)

#-------------------------------------------------------------------------------
class ARFirstTrueBAxis0Forward(ArrayProcessor):
NAME = 'ar.first_true_2d_b(forward=True, axis=0)'
SORT = 20

def __call__(self):

_ = ar_first_true_2d_b(self.array, forward=True, axis=0)

class ARFirstTrueBAxis1Forward(ArrayProcessor):
NAME = 'ar.first_true_2d_b(forward=True, axis=1)'
SORT = 21

def __call__(self):
_ = ar_first_true_2d_b(self.array, forward=True, axis=1)

class ARFirstTrueBAxis0Reverse(ArrayProcessor):
NAME = 'ar.first_true_2d_b(forward=False, axis=0)'
SORT = 22

def __call__(self):
_ = ar_first_true_2d_b(self.array, forward=False, axis=0)

class ARFirstTrueBAxis1Reverse(ArrayProcessor):
NAME = 'ar.first_true_2d_b(forward=False, axis=1)'
SORT = 23

def __call__(self):
_ = ar_first_true_2d_b(self.array, forward=False, axis=1)



#-------------------------------------------------------------------------------


class NPNonZero(ArrayProcessor):
NAME = 'np.nonzero()'
SORT = 3
SORT = 33

def __call__(self):
x, y = np.nonzero(self.array)
Expand All @@ -101,15 +134,15 @@ def __call__(self):

class NPArgMaxAxis0(ArrayProcessor):
NAME = 'np.any(axis=0), np.argmax(axis=0)'
SORT = 4
SORT = 34

def __call__(self):
_ = ~np.any(self.array, axis=0)
_ = np.argmax(self.array, axis=0)

class NPArgMaxAxis1(ArrayProcessor):
NAME = 'np.any(axis=1), np.argmax(axis=1)'
SORT = 4
SORT = 35

def __call__(self):
_ = ~np.any(self.array, axis=1)
Expand Down Expand Up @@ -303,14 +336,20 @@ def get_versions() -> str:
AKFirstTrueAxis0Reverse,
AKFirstTrueAxis1Reverse,

ARFirstTrueAxis0Forward,
ARFirstTrueAxis1Forward,
ARFirstTrueAxis0Reverse,
ARFirstTrueAxis1Reverse,
ARFirstTrueAAxis0Forward,
ARFirstTrueAAxis1Forward,
ARFirstTrueAAxis0Reverse,
ARFirstTrueAAxis1Reverse,

ARFirstTrueBAxis0Forward,
ARFirstTrueBAxis1Forward,
ARFirstTrueBAxis0Reverse,
ARFirstTrueBAxis1Reverse,


# NPNonZero,
# NPArgMaxAxis0,
# NPArgMaxAxis1
NPArgMaxAxis0,
NPArgMaxAxis1
)

CLS_FF = (
Expand Down
Loading