3
2

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 3 years have passed since last update.

whereメソッドの統一感がない残念なお知らせ

Posted at

numpyとpandasのwhereの挙動が反するように感じたので、共有します。

条件に応じて値を代入するメソッドにwheremaskがあります。
メソッドの詳細はこちらにございます。
(いつもお世話になっております:pray:

以下に適当な例を出してみます。

import numpy as np
import pandas as py

test = np.array([1,2,3])

この2200に置換したいとします。

np.whereでやってみましょう。

>>> pd.Series(np.where(test == 2, 200, test))
0      1
1    200
2      3
dtype: int64

目的の結果が得られます。

では、pd.whereでは?

>>> pd.Series(test).where(test == 2, 200)
0    200
1      2
2    200
dtype: int64

残念です。目的とは逆の結果になりました…

np.whereの代わりにはpd.maskを使いましょう。

>>> pd.Series(np.where(test == 2, 200, test))
0      1
1    200
2      3
dtype: int64

np.whereとの挙動を合わせてほしいですね…

3
2
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
3
2

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?