How to run variable of one function to another in python?-CodePudding

I am a beginner. I have two function: 1st creating dataframes and some print statement 2nd is downloading the dataframes to csv in colab.

I want to download all dataframes by the df_name. code:

def fun1():
  import pandas as pd
  d = {'col1': [1, 2], 'col2': [3, 4]}
  d2 = {'col1': [-5, -6], 'col2': [-7, -8]}
  df = pd.DataFrame(data=d)
  df2 = pd.DataFrame(data=d2)
  print('info', df.info())
  print('info', df2.info())
  return df, df2
def fun2(df):
  from google.colab import files
  name1 = 'positive.csv'
  name2 = 'negative.csv'
  df.to_csv(name1)
  df2.to_csv(name2)
  files.download(name1)
  files.download(name2)

fun2(df) #looking something like this that download my df, func2 should read my df and df2 from fun1()

I tried:

class tom:
  def fun1(self):
    import pandas as pd
    d = {'col1': [1, 2], 'col2': [3, 4]}
    d2 = {'col1': [-5, -6], 'col2': [-7, -8]}
    df = pd.DataFrame(data=d)
    df2 = pd.DataFrame(data=d2)
    print('info', df.info())
    print('info', df2.info())
    self.df= df
    self.df2 = df2
    
    return df, df2
 
  def fun2(self):
    df,df2 = fun1()
    from google.colab import files
    name1 = 'positive.csv'
    name2 = 'negative.csv'
    df.to_csv(name1)
    df2.to_csv(name2)
    return files.download(name1) ,files.download(name2)

tom().fun2() #it download files but shows print of fun1 as well which I don't want.

looking for something like

tom().fun2(dataframe_name) #it just download the files nothing else

CodePudding user response：

set permanent variables directly in the class if its not gonna change and

define fun just for actions.

class s:
  import pandas as pd
  
  d = {'col1': [1, 2], 'col2': [3, 4]}
  d2 = {'col1': [-5, -6], 'col2': [-7, -8]}
  df = pd.DataFrame(data=d)
  df2 = pd.DataFrame(data=d2)


  name1 = 'positive.csv'
  name2 = 'negative.csv'
  df.to_csv(name1)
  df2.to_csv(name2)
  def f():
    print('info', df.info())
    print('info', df2.info())
  
  def fun(x):
    from google.colab import files
    return files.download(x)

run

s.f() --it will print value only
s.fun(name1) --it will just download the file

CodePudding user response：

Maybe you can save the data you need in a class variable or create another function, that keeps the data from the first function you need the value (lets call it A) and then pass A to the second function as an argument.