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

python

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

例えば、以下のような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に対して、mask()メソッドを使用して、A列の値が2以上の場合にだけ、特定の値を保持するようにします。以下のようなコードを書きます。

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

実行結果:

   A  B   C
0 0 0 0
1 2 6 10
2 3 7 11
3 4 8 12

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

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