Jak sprawdzić, czy jesteś w notebooku Jupyter

Question

Tworzę moduł python z funkcją wyświetlania pandas DataFrame (my_df).

Jeśli użytkownik importuje moduł do notebooka Jupyter, chciałbym dostarczyć „ładny” formatowanie dla DataFrame za pomocą coś jak:

from IPython.display import display, HTML 
display(my_df) 

Jeśli użytkownik nie jest w zeszycie Jupyter, ja chciałbym wyświetlić formularz tekstowy DataFrame:

print(my_df) 

Jak mogę sprawdzić, czy kod jest uruchamiany z notatnika Jupyter? Albo, jak mogę wyświetlić DataFrame w formie tekstowej z linii poleceń, a wyświetlić formularz HTML, jeśli jest on importowany do notatnika Jupyter?

from IPython.display import display, HTML 

def my_func(my_df): 
    if [... code to check for Jupyter notebook here ...]: 
     display(my_df) 
    else: 
     print(my_df) 

Answer 1

Nie trzeba sprawdzić, czy kod jest uruchamiany z notebooka; display() drukuje tekst po wywołaniu z wiersza poleceń.

test.py:

from IPython.display import display 
import pandas as pd 

my_df = pd.DataFrame({'foo':[1,2,3],'bar':[7,8,9]}) 
display(my_df) 

Z wiersza poleceń:

$ python test.py 
    bar foo 
0 7 1 
1 8 2 
2 9 3 

z notebooka Jupyter:

UPDATE
Aby sprawdzić, czy używasz interaktywnej powłoki Ipython (w wierszu polecenia lub w przeglądarce), sprawdź numer get_ipython. (Zaadaptowane z Ipython docs)

Modified test.py:

from IPython.display import display, HTML 
import pandas as pd 
my_df = pd.DataFrame({'foo':[1,2,3],'bar':[7,8,9]}) 

try: 
    get_ipython 
    display(my_df) 
except: 
    print(my_df) 

Takie podejście:
- całkiem-print w przeglądarce Jupyter notebook
- tekst do druku, gdy uruchamiane jako skrypt z linii poleceń (np. python test.py) - jeśli uruchomi się wiersz po linii w powłoce Pythona, nie zamieni się w interaktywną powłokę Ipython po wydrukowaniu

Answer 2

Powinieneś wyglądać w os.environ.

Na moim komputerze można zobaczyć go w

os.environ['_']