Python 함수 호출에서 표준 출력 캡처
객체를 수정하고 통계를 stdout으로 인쇄하는 Python 라이브러리를 사용할 때 다음이 필요할 수 있습니다. 추가 분석을 위해 이 출력을 캡처합니다. 그러나 이러한 정보를 반환하도록 함수를 직접 수정하는 것이 항상 가능한 것은 아닙니다.
이 문제를 해결하려면 캡처 컨텍스트 관리자를 활용할 수 있습니다.
<code class="python">from io import StringIO import sys class Capturing(list): def __enter__(self): self._stdout = sys.stdout sys.stdout = self._stringio = StringIO() return self def __exit__(self, *args): self.extend(self._stringio.getvalue().splitlines()) del self._stringio # free up some memory sys.stdout = self._stdout</code>
캡처 컨텍스트 관리자를 사용할 수 있습니다. 다음과 같습니다:
<code class="python">with Capturing() as output: do_something(my_object)</code>
함수 호출 후 출력 목록에는 함수에 의해 인쇄된 줄이 포함됩니다.
이 기술은 여러 번 적용할 수 있으며 결과는 연결될 수 있습니다.
<code class="python">with Capturing() as output: print('hello world') print('displays on screen') with Capturing(output) as output: # note the constructor argument print('hello world2') print('done') print('output:', output)</code>
출력:
displays on screen done output: ['hello world', 'hello world2']
이 접근 방식은 do_something() 함수를 직접 수정할 수 없는 경우 stdout 출력을 캡처하기 위한 효과적인 해결 방법을 제공합니다.
위 내용은 Python 함수 호출에서 표준 출력을 캡처하는 방법은 무엇입니까?의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!