pandas equivalent of np.where

by

Try:

(df['A'] + df['B']).where((df['A'] < 0) | (df['B'] > 0), df['A'] / df['B'])

The difference between the numpy where and DataFrame where is that the default values are supplied by the DataFrame that the where method is being called on (docs).

I.e.

np.where(m, A, B)

is roughly equivalent to

A.where(m, B)

If you wanted a similar call signature using pandas, you could take advantage of the way method calls work in Python:

pd.DataFrame.where(cond=(df['A'] < 0) | (df['B'] > 0), self=df['A'] + df['B'], other=df['A'] / df['B'])

or without kwargs (Note: that the positional order of arguments is different from the numpy where argument order):

pd.DataFrame.where(df['A'] + df['B'], (df['A'] < 0) | (df['B'] > 0), df['A'] / df['B'])

More Related Contents:

  1. Still getting ModuleNotFoundError: No module named ‘pandas’ ‘ despite having installed Pandas, anaconda, [closed]
  2. Binning a column with Python Pandas
  3. Best way to join / merge by range in pandas
  4. How to take column-slices of dataframe in pandas
  5. Creating a Pandas DataFrame from a Numpy array: How do I specify the index column and column headers?
  6. float64 with pandas to_csv
  7. How to check if a value is in the list in selection from pandas data frame?
  8. Replacing Pandas or Numpy Nan with a None to use with MysqlDB
  9. shuffling/permutating a DataFrame in pandas
  10. Pandas finding local max and min
  11. SQL-like window functions in PANDAS: Row Numbering in Python Pandas Dataframe
  12. Recursive definitions in Pandas
  13. extracting days from a numpy.timedelta64 value
  14. Set values on the diagonal of pandas.DataFrame
  15. What techniques can be used to measure performance of pandas/numpy solutions
  16. how do I calculate a rolling idxmax
  17. binning a dataframe in pandas in Python [duplicate]
  18. Custom transformer for sklearn Pipeline that alters both X and y
  19. Python Numpy TypeError: ufunc ‘isfinite’ not supported for the input types
  20. Pass percentiles to pandas agg function
  21. Why is np.where faster than pd.apply
  22. How to determine whether a column/variable is numeric or not in Pandas/NumPy?
  23. pandas.read_csv from string or package data
  24. How to automatically annotate maximum value in pyplot
  25. A faster alternative to Pandas `isin` function
  26. Split Python sequence (time series/array) into subsequences with overlap
  27. Faster way to transform group with mean value in Pandas
  28. Add density curve on the histogram
  29. Bin values based on ranges with pandas [duplicate]
  30. Numpy.dtype has the wrong size, try recompiling

Leave a Comment