Get column name in data frame if value is between values

Question

I have a dataframe:

import numpy as np
import pandas as pd

random_number_gen = np.random.default_rng()
df = pd.DataFrame(random_number_gen.integers(-5, 5, size=(1, 13)), columns=list('ABCDEFGHIJKLM'))

A	B	C	D	E	F	G	H	I	J	K	L	M
0	1	4	-4	-1	3	-5	-3	0	-4	-1	3	2

I would like to obtain the names of the columns where a value falls between -1 and 1. I tried this and others:

df.columns[(( -1<= df.any()) & (df.any() <=1)).iloc[0]]

Any help is welcome. Thanks.

Answer 1

If you have a single row:

df.columns[df.iloc[0].between(-1,1)]

# or
df.columns[df.squeeze().between(-1,1)]

If you can have multiple rows:

df.columns[(df.ge(-1)&df.le(1)).any()]

Example output:

Index(['E', 'G', 'J'], dtype='object')

Used input:

   A  B  C  D  E  F  G  H  I  J  K  L  M
0  3 -3 -4 -3 -1  3 -1 -5 -2  1  3  2  4