pandas.DataFrame.where

DataFrame.where(cond, other=nan, inplace=False, axis=None, level=None, errors='raise', try_cast=False, raise_on_error=None) [source]

Возвращает объект той же формы, что и self, чьи соответствующие элементы взяты из self, где cond истинно, а в противном случае — из other.

Параметры:

cond : булевый NDFrame, массив или вызываемый объект Когда cond истинно, сохраняется исходное значение. Когда ложно, заменяется соответствующим значением из other. Если cond — вызываемый объект, он вычисляется для NDFrame и должен возвращать булевый NDFrame или массив. Вызываемый объект не должен изменять входной NDFrame (хотя pandas этого не проверяет). Добавлена в версии 0.18.1: В качестве cond можно использовать вызываемый объект. other : скаляр, NDFrame или вызываемый объект Элементы, где cond ложно, заменяются соответствующим значением из other. Если other — вызываемый объект, он вычисляется для NDFrame и должен возвращать скаляр или NDFrame. Вызываемый объект не должен изменять входной NDFrame (хотя pandas этого не проверяет). Добавлена в версии 0.18.1: В качестве other можно использовать вызываемый объект. inplace : булево, по умолчанию False Выполнять ли операцию на месте с данными axis : ось выравнивания при необходимости, по умолчанию None level : уровень выравнивания при необходимости, по умолчанию None errors : строка, {‘raise’, ‘ignore’}, по умолчанию ‘raise’ raise : разрешить исключения ignore : подавить исключения. В случае ошибки возвращается исходный объект Обратите внимание, что в настоящее время этот параметр не повлияет на результаты и всегда будет приводиться к подходящему типу данных. try_cast : булево, по умолчанию False попытаться преобразовать результат обратно к типу входных данных (если возможно) raise_on_error : булево, по умолчанию True Вызывать ли исключение при недопустимых типах данных (например, при попытке where со строками) Устарело начиная с версии 0.21.0.

Возвращает:

wh : тот же тип, что и вызываемый объект

См. также

DataFrame.mask()

Примечания

Метод where — это применение конструкции «если-то». Для каждого элемента в вызываемом DataFrame, если cond является True, используется элемент; в противном случае используется соответствующий элемент из DataFrame other.

Подпись DataFrame.where() отличается от numpy.where(). Примерно df1.where(m, df2) эквивалентно np.where(m, df1, df2).

Дополнительные сведения и примеры см. в where документации в индексации.

Примеры

>>> s = pd.Series(range(5))
>>> s.where(s > 0)
0    NaN
1    1.0
2    2.0
3    3.0
4    4.0

>>> s.mask(s > 0)
0    0.0
1    NaN
2    NaN
3    NaN
4    NaN

>>> s.where(s > 1, 10)
0    10.0
1    10.0
2    2.0
3    3.0
4    4.0

>>> df = pd.DataFrame(np.arange(10).reshape(-1, 2), columns=['A', 'B'])
>>> m = df % 3 == 0
>>> df.where(m, -df)
   A  B
0  0 -1
1 -2  3
2 -4 -5
3  6 -7
4 -8  9
>>> df.where(m, -df) == np.where(m, df, -df)
      A     B
0  True  True
1  True  True
2  True  True
3  True  True
4  True  True
>>> df.where(m, -df) == df.mask(~m, -df)
      A     B
0  True  True
1  True  True
2  True  True
3  True  True
4  True  True