[python] Pandas에서 숫자 열을 어떻게 찾나요?

dfpandas DataFrame 이라고 가정 해 보겠습니다 . 숫자 유형의 모든 열을 찾고 싶습니다. 다음과 같은 것 :

isNumeric = is_numeric(df)

답변

select_dtypesDataFrame의 방법을 사용할 수 있습니다 . 여기에는 두 개의 매개 변수 include 및 exclude가 포함됩니다. 따라서 isNumeric은 다음과 같습니다.

numerics = ['int16', 'int32', 'int64', 'float16', 'float32', 'float64']

newdf = df.select_dtypes(include=numerics)

답변

문서화되지 않은 함수 _get_numeric_data()를 사용하여 숫자 열만 필터링 할 수 있습니다 .

df._get_numeric_data()

예:

In [32]: data
Out[32]:
   A  B
0  1  s
1  2  s
2  3  s
3  4  s

In [33]: data._get_numeric_data()
Out[33]:
   A
0  1
1  2
2  3
3  4

이는 “비공개 방법”(즉, 구현 세부 사항)이며 향후 변경되거나 완전히 제거 될 수 있습니다. 주의해서 사용하십시오 .

답변

숫자 열만있는 새 데이터 프레임을 만드는 간단한 한 줄 답변 :

df.select_dtypes(include=np.number)

숫자 열의 이름을 원하는 경우 :

df.select_dtypes(include=np.number).columns.tolist()

완전한 코드 :

import pandas as pd
import numpy as np

df = pd.DataFrame({'A': range(7, 10),
                   'B': np.random.rand(3),
                   'C': ['foo','bar','baz'],
                   'D': ['who','what','when']})
df
#    A         B    C     D
# 0  7  0.704021  foo   who
# 1  8  0.264025  bar  what
# 2  9  0.230671  baz  when

df_numerics_only = df.select_dtypes(include=np.number)
df_numerics_only
#    A         B
# 0  7  0.704021
# 1  8  0.264025
# 2  9  0.230671

colnames_numerics_only = df.select_dtypes(include=np.number).columns.tolist()
colnames_numerics_only
# ['A', 'B']

답변

df.select_dtypes(exclude=['object'])

답변

간단한 한 줄 :

df.select_dtypes('number').columns

답변

다음 코드는 데이터 세트의 숫자 열 이름 목록을 리턴합니다.

cnames=list(marketing_train.select_dtypes(exclude=['object']).columns)

여기 marketing_train에 내 데이터 세트가 있으며 select_dtypes()제외 및 포함 인수를 사용하여 데이터 유형을 선택하는 기능이며 위 코드의 데이터 세트 출력의 열 이름을 가져 오는 데 사용되는 열은 다음과 같습니다.

['custAge',
     'campaign',
     'pdays',
     'previous',
     'emp.var.rate',
     'cons.price.idx',
     'cons.conf.idx',
     'euribor3m',
     'nr.employed',
     'pmonths',
     'pastEmail']

감사

답변

이것은 pandas 데이터 프레임에서 숫자 열을 찾는 또 다른 간단한 코드입니다.

numeric_clmns = df.dtypes[df.dtypes != "object"].index