Pythonの条件抽出に関しまして

Question

実現したいこと

以下のようなデータフレームがあって、数値の0.2以上、−０.２以下の数値を抽出し
0.2以上、-0.2以下の数値が一つでも残っている行は、全ての値を残し、一つも残っていない行は削除したいです。どなたか、ご指導お願いいたします。

該当のソースコード

ソースコード

df = pd.read_csv('data/src/sample_pandas_normal.csv')

print(df)
name A B C
0 Alice 0.1 0.3 0.05
1 Bob 0.3 -0.5 -0.02
2 Charlie 0.01 -0.1 -0.01
3 Dave 0.1 0.05 -0.05
4 Ellen 0.02 0.03 0.5
5 Frank 0.8 0.02 0.3

結果的に数値の0.2以上、−０.２以下の数値を抽出し、0.2以上、-0.2以下の数値が一つでも残っている行は、全ての値を残し、一つも残っていない行は削除したいです。このように輩出したいです。

print(df)
name A B C
0 Alice 0.1 0.3 0.05
1 Bob 0.3 -0.5 -0.02
4 Ellen 0.02 0.03 0.5
5 Frank 0.8 0.02 0.3

Answer 1

以下のような感じでできます。

import pandas as pd
from io import StringIO

s = """name A B C
Alice 0.1 0.3 0.05
Bob 0.3 -0.5 -0.02
Charlie 0.01 -0.1 -0.01
Dave 0.1 0.05 -0.05
Ellen 0.02 0.03 0.5
Frank 0.8 0.02 0.3"""
df = pd.read_csv(StringIO(s),sep='\s+')

# name列を除き、行方向に絶対値が指定値以上の１つでも要素がある
cond = (df.iloc[:,1:].abs() >= 0.2).any(axis=1)
print(cond)
"""
0     True
1     True
2    False
3    False
4     True
5     True
"""

print(df[cond])
"""
    name     A     B     C
0  Alice  0.10  0.30  0.05
1    Bob  0.30 -0.50 -0.02
4  Ellen  0.02  0.03  0.50
5  Frank  0.80  0.02  0.30
"""

Answer 2

import pandas as pd
import numpy as np

df = pd.read_csv('data/src/sample_pandas_normal.csv')

idx = df.select_dtypes(include=[np.number]).abs().ge(0.2).any(axis=1)
dfx = df[idx].reset_index(drop=True)

print(dfx)

name	A	B	C
Alice	0.1	0.3	0.05
Bob	0.3	-0.5	-0.02
Ellen	0.02	0.03	0.5
Frank	0.8	0.02	0.3

Comments

[Ryosuke0524]

ありがとうございます！こちらで実行できました。とても助かりました。