【Python】in 검색한 결과의 컬럼만 표시시킨다
경위
요전날 시험으로 만든 함수가 상당히 편리할까라고 생각했으므로 메모 ✍️
250컬럼 정도 있는 가운데에서 "유통기한xx"와 같은 컬럼만 추출했다
하고 싶은 일
df[["賞味期限1", "賞味期限2",:...]]
같은 수치가 줄어들 👈 여기를 개선하고 싶었다 만든 함수
# in 検索した結果のカラムのみ表示させる
def in_search_col_list(df, cols="賞味期限", head=2) -> List:
""" in search した colの結果を返す
params
----------
df(DataFrame): 取り込みデータ
cols(str): 検索にかけたいグローバルの値
head(int): 先頭から抽出するレコード数
return
----------
df(IndexList) in search したカラムリスト結果を返す
"""
return df[[for col in list(df.columns) if cols in col]].head(head)
사용 예
# 必要package
import pandas as pd
import numpy as np
from typing import List
# サンプルデータ(実際は, *tabファイルを読み込んでいたりします。)
df = pd.DataFrame({ '賞味期限A' : 1.,
'保管場所B' : pd.Timestamp('20130102'),
'保管場所C' : pd.Series(1,index=list(range(5)),dtype='float32'),
'賞味期限D' : np.array([3] * 5,dtype='int32'),
'E賞味期限' : pd.Categorical(["test","train","test","train", "test"]),
'食品個F' : 'foo',
'期間限定G' : ['1000', np.nan, '123', '1234', '']})
def in_search_col_list(df, cols="賞味期限", head=2) -> List:
return df[[ col for col in list(df.columns) if cols in col]].head(head)
# 関数呼び出し
in_search_col_list(df, cols="賞味期限", head=2)
출력 결과
요약
# in 検索した結果のカラムのみ表示させる
def in_search_col_list(df, cols="賞味期限", head=2) -> List:
""" in search した colの結果を返す
params
----------
df(DataFrame): 取り込みデータ
cols(str): 検索にかけたいグローバルの値
head(int): 先頭から抽出するレコード数
return
----------
df(IndexList) in search したカラムリスト結果を返す
"""
return df[[for col in list(df.columns) if cols in col]].head(head)
# 必要package
import pandas as pd
import numpy as np
from typing import List
# サンプルデータ(実際は, *tabファイルを読み込んでいたりします。)
df = pd.DataFrame({ '賞味期限A' : 1.,
'保管場所B' : pd.Timestamp('20130102'),
'保管場所C' : pd.Series(1,index=list(range(5)),dtype='float32'),
'賞味期限D' : np.array([3] * 5,dtype='int32'),
'E賞味期限' : pd.Categorical(["test","train","test","train", "test"]),
'食品個F' : 'foo',
'期間限定G' : ['1000', np.nan, '123', '1234', '']})
def in_search_col_list(df, cols="賞味期限", head=2) -> List:
return df[[ col for col in list(df.columns) if cols in col]].head(head)
# 関数呼び出し
in_search_col_list(df, cols="賞味期限", head=2)
요약
cols
의 인수명에 검색하고 싶은 캐릭터 라인을 넣으면, 한발로 검색할 수 있으므로 편리Reference
이 문제에 관하여(【Python】in 검색한 결과의 컬럼만 표시시킨다), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://qiita.com/KazusaNakagawa/items/42e1db34f9d59c317f41텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)