如果值在值之間，則獲取數據框中的列名

Question

我有一個 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'))

一個	乙	C	D	乙	F	G	H	我	Ĵ	ķ	大號	米
0	1	4	-4	-1	3	-5	-3	0	-4	-1	3	2

我想獲取值介於 -1 和 1 之間的列的名稱。我試過這個和其他的：

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

歡迎任何幫助。 謝謝。

Answer 1

如果您只有一行：

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

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

如果您可以有多行：

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

示例 output：

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

使用的輸入：

   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