【Pandas DataFrame】where()メソッドで、データフレームの各要素が特定の条件を満たす場合にだけ、特定の値を保持するようにすることができます。

python

PandasのDataFrameには、where()メソッドという機能があります。これは、データフレームの各要素が特定の条件を満たす場合にだけ、特定の値を保持するようにするための機能です。

例えば、以下のようなDataFrameを考えます。

import pandas as pd
df = pd.DataFrame({'A': [1, 2, 3, 4],
'B': [5, 6, 7, 8],
'C': [9, 10, 11, 12]})
print(df)
   A  B   C
0 1 5 9
1 2 6 10
2 3 7 11
3 4 8 12

このDataFrameに対して、where()メソッドを使うと、以下のように、特定の条件を満たす場合にだけ、特定の値を保持するようにすることができます。

df.where(df > 5, 0)
   A  B  C
0 0 5 0
1 0 6 0
2 3 7 0
3 4 8 12

この例では、DataFrameの各要素が5より大きい場合はそのまま、5以下の場合は0に置き換えています。

タイトルとURLをコピーしました