pandas 和 numpy 中 where 使用

注意： df1.where(cond,df2) 等价于 np.where(cond, df1, df2)

1. pandas.DataFrame.where

DataFrame.where(cond, other=nan, inplace=False, axis=None, level=None, try_cast=False, raise_on_error=True)
功能：按条件查找替换，cond 为 True 则 self 值保持不变；False 改为参数 other 对应的值。结果返回与 self 相同的对象。
参数：
cond 查找条件
other cond为False时要替换的值
inplace 是否在原数据上操作

>>> import numpy as np
>>> import pandas as pd
>>> data = pd.DataFrame(np.arange(12).reshape(3,4),columns=['a','b','c','d'])
>>> data
   a  b   c   d
0  0  1   2   3
1  4  5   6   7
2  8  9  10  11
>>> data.where(data['a']>3, 3.0)
   a  b   c   d
0  3  3   3   3
1  4  5   6   7
2  8  9  10  11

2.numpy.where

声明： 首先强调一下，where()函数对于不同的输入，返回值是不同的。

1. 当数组是一维数组时，返回的值是一维的索引，所以只有一组索引数组

2. 当数组是多维数组时，满足条件的数组值返回的是值的位置索引，因此会有两组索引数组来表示值的位置。

>>> import numpy as np
>>> a = np.arange(10)
>>> a
array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> np.where(a>5)
(array([6, 7, 8, 9], dtype=int64),)

>>> b = np.arange(20).reshape(4,5)
>>> b
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]])
>>> np.where(b>10)
(array([2, 2, 2, 2, 3, 3, 3, 3, 3], dtype=int64), array([1, 2, 3, 4, 0, 1, 2, 3, 4], dtype=int64))

具体实现

numpy.where(condition[, x, y])
功能：
参数：
condition： 判定条件，如果True，选择 x；False，选择y（数据类型为数组，bool 值）x,y（可选）： x 和 y 的 shape 必须和 condition 相同（可以采用 broadcast，广播机制）
①如果参数有condition，x和y，它们三个参数的shape是相同的。那么，当condition中的值是true时返回x对应位置的值，false是返回y的。
②如果参数只有condition的话，返回值是condition中元素值为true的位置索引，且是以元组形式返回，元组的元素是ndarray数组，表示位置的索引

>>> np.where([[True, False], [True, True]],	# 官网例子
...          [[1, 2], [3, 4]],
...          [[9, 8], [7, 6]])
array([[1, 8],
       [3, 4]])
       
 >>> x = np.arange(9).reshape(3, 3)
>>> np.where(x>5)		# ndarray 数组分别表示对应的 行和列
(array([2, 2, 2], dtype=int64), array([0, 1, 2], dtype=int64))

https://www.cnblogs.com/zz22–/p/8654395.html