Python при ведении журнала не регистрируется pd.info () - PullRequest
1 голос
/ 04 апреля 2020
import logging
import pandas as pd

logger = logging.getLogger('train')
logger.setLevel(logging.DEBUG)

# Data
data = {'Name': ['Tom', 'nick', 'krish', 'jack'], 'Age': [20, 21, 19, 18]}

# Create DataFrame
df = pd.DataFrame(data)

logger.info(type(df))
logger.info(df.info())
.
.
.
<other_processes>
.

Приведенный выше код выводит:

<class 'pandas.core.frame.DataFrame'>
None
.
.
.

И в конце журналов (после всех других процессов) он также выводит:

<class 'pandas.core.frame.DataFrame'>
RangeIndex: 4 entries, 0 to 3
Data columns (total 2 columns):
Name    4 non-null object
Age     4 non-null int64
dtypes: int64(1), object(1)
memory usage: 144.0+ bytes

Почему это печатать None когда я пытаюсь войти в систему df.info ()? Как я могу получить df.info () в нужном месте в моих журналах?

1 Ответ

1 голос
/ 04 апреля 2020

Изменить параметр буфера в DataFrame.info на StringIO для текста с .getvalue():

from io import StringIO
buf = StringIO()
df.info(buf=buf)

logger.info(type(df))    
logger.info(buf.getvalue())
...