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


121

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

isNumeric = is_numeric(df)

당신은 한 열 것인지 여부를 지정해야 dtype되고 object있지만, 모든 요소가 숫자 인 숫자 또는하지 계산됩니다. 그렇지 않다면 Hanan의 대답을 취하십시오. 그렇지 않으면 내 것을 가져 가라.
FooBar

단순히 df.describe (). columns를 시도하면 어떻게됩니까? 그런 다음 변수에 할당하십시오.
coldy

답변:


146

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

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

newdf = df.select_dtypes(include=numerics)

94
'숫자'목록을 지정할 필요가없는 경우 df.select_dtypes (include = [np.number])를 사용할 수 있습니다
KieranPC

23
이전의 주석 (+1)의 팁에 구축, 당신은 사용할 수있는 list(df.select_dtypes(include=[np.number]).columns.values) 숫자 열 이름의 목록을 얻을
user799188

76

문서화되지 않은 함수 _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

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


1
매우 편리합니다. 문서화되어 있습니까? 접두사 밑줄은 개인용임을 나타 내기 때문에
ijoseph

3
아니요, 이것은 어디에도 문서화되어 있지 않습니다. 구현은 여기 에 있지만 @ijoseph가 언급했듯이 밑줄로 시작하는 메서드는 구현 세부 사항에 지나지 않으므로주의해야합니다. 이 외에 말 그대로 다른 대답을 사용하십시오.
cs95

바로 그거죠. 모범 사례로 가능한 한 많은 numpy 메서드를 사용하고 변환하려고합니다. 이것은 판다의 역 동성 때문입니다. API는 자주 변경됩니다. 문서화되지 않은 방법의 경우 아무리 유용하더라도 무모합니다.
mik

69

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

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']

2
df.select_dtypes(include=['int64']).columns.tolist()
Cherry Wu

한 가지 유형 만 원하는 경우 목록에 저장할 필요가 없습니다. 을 지정할 필요도 없습니다 include=. select_dtypes(np.number)
BallpointBen

열에 숫자 데이터가 있지만 None도있는 경우 dtype은 'object'가 될 수 있습니다. 그러면 열이 숫자로 강제 변환됩니다.df.fillna(value=0, inplace=True)
vaughnkoch


15

간단한 한 줄 :

df.select_dtypes('number').columns

2
지금까지 가장 파이썬적인 방법입니다.
jorijnsmit

6

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

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']

감사


4

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

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

1
def is_type(df, baseType):
    import numpy as np
    import pandas as pd
    test = [issubclass(np.dtype(d).type, baseType) for d in df.dtypes]
    return pd.DataFrame(data = test, index = df.columns, columns = ["test"])
def is_float(df):
    import numpy as np
    return is_type(df, np.float)
def is_number(df):
    import numpy as np
    return is_type(df, np.number)
def is_integer(df):
    import numpy as np
    return is_type(df, np.integer)

1

이 답변을 적용 하면 할 수 있습니다.

df.ix[:,df.applymap(np.isreal).all(axis=0)]

여기 np.applymap(np.isreal)에서는 데이터 프레임의 모든 셀이 숫자인지 여부를 보여주고 .axis(all=0)열의 모든 값이 True인지 확인하고 원하는 열을 인덱싱하는 데 사용할 수있는 일련의 부울을 반환합니다.


1

아래 코드를 참조하십시오.

if(dataset.select_dtypes(include=[np.number]).shape[1] > 0):
display(dataset.select_dtypes(include=[np.number]).describe())
if(dataset.select_dtypes(include=[np.object]).shape[1] > 0):
display(dataset.select_dtypes(include=[np.object]).describe())

이렇게하면 값이 float 및 int 또는 srting 값과 같은 숫자인지 확인할 수 있습니다. 두 번째 if 문은 객체가 참조하는 문자열 값을 확인하는 데 사용됩니다.


1

다음과 같은 요구 사항에 따라 데이터 유형을 포함 및 제외 할 수 있습니다.

train.select_dtypes(include=None, exclude=None)
train.select_dtypes(include='number') #will include all the numeric types

Jupyter Notebook에서 참조했습니다.

모든 숫자 유형 을 선택하려면 np.number또는'number'

  • 문자열을 선택하려면 objectdtype을 사용해야하지만 모든 개체 dtype 열을 반환 합니다.

  • NumPy dtype hierarchy <http://docs.scipy.org/doc/numpy/reference/arrays.scalars.html>__ 참조

  • 날짜 / 시간을 선택하려면 np.datetime64, 'datetime'또는 'datetime64'

  • 타임 델타를 선택하려면 np.timedelta64, 'timedelta'또는 'timedelta64'

  • Pandas 범주 형 dtype을 선택하려면 'category'

  • Pandas datetimetz dtypes를 선택하려면 'datetimetz'( 0.20.0의 새로운 기능) 또는`` 'datetime64 [ns, tz]'를 사용하십시오.

당사 사이트를 사용함과 동시에 당사의 쿠키 정책개인정보 보호정책을 읽고 이해하였음을 인정하는 것으로 간주합니다.
Licensed under cc by-sa 3.0 with attribution required.