Retour d'un dataframe en fonction python

Je suis en train de créer et retourner d'un bloc de données à partir d'une fonction python 

def create_df():
    data = {'state': ['Ohio','Ohio','Ohio','Nevada','Nevada'],
           'year': [2000,2001,2002,2001,2002],
           'pop': [1.5,1.7,3.6,2.4,2.9]}
    df = pd.DataFrame(data)
    return df
create_df()
df

J'obtiens une erreur que df n'est pas défini. Si je remplace le "retour" par "imprimer" - je obtenir de l'impression de la trame de données correctement. Est-il un moyen de faire cela? grâce

  • df est une variable locale. Vous devez affecter le résultat comme df = create_df().
InformationsquelleAutor Manoj Agrawal | 2017-08-08
  1. 7

    lorsque vous appelez create_df() python appelle la fonction, mais n'a pas d'enregistrer le résultat dans une variable. c'est pourquoi vous avez l'erreur.

    affecter le résultat de create_df() à df comme ce df = create_df()

    • aarrgh...merci!
    InformationsquelleAutor OLIVER.KOO
  2. 2

    Je suis un peu en retard, mais ce que sur la création d'une variable globale au sein de la fonction? Il doit enregistrer une étape pour vous.

    def create_df():

    global df

    data = {
    'state': ['Ohio','Ohio','Ohio','Nevada','Nevada'],
    'year': [2000,2001,2002,2001,2002],
    'pop': [1.5,1.7,3.6,2.4,2.9]
    }

    df = pd.DataFrame(data)

    Puis lorsque vous exécutez create_df(), vous serez en mesure de simplement utiliser df.

    Bien sûr, être prudent dans votre stratégie de nommage si vous avez un grand programme de sorte que la valeur de df ne change pas que les diverses fonctions de s'exécuter.

    InformationsquelleAutor Mark Moretto
  3. 0

    Fonction renvoie explicitement deux DataFrames:

    import pandas as pd
import numpy as np

def return_2DF():

date = pd.date_range('today', periods=20)
DF1 = pd.DataFrame(np.random.rand(20, 2), index=date, columns=list('xyz'))

DF2 = pd.DataFrame(np.random.rand(20, 4), index=date, columns='A B C D'.split())

return DF1, DF2

    L'appel et le retour de deux trames de données 

    one, two = return_2DF()
    InformationsquelleAutor Ram Prajapati