Как добавить строку в качестве префикса при именовании фрейма данных - PullRequest
0 голосов
/ 09 мая 2020

У меня есть функция, которая принимает два фрейма данных и строку, как показано ниже, но я не могу понять, как создать новый фрейм данных, где я хочу, чтобы имя фрейма данных было строкой, которую я ввожу в функцию конкатенирован с «местом_файла» в конце.

def basic_pivot(df1,df2,str):
    if str=='first':
       str+'venue_df'=df1.pivot_table(index=['venue_id','max'],
                                    values=['match','name','continent','country'],
                                    aggfunc={'match':'nunique','name':'last',
                                             'continent':'first','country':'first'})
    else:
       None
 return str+'venue_df'

1 Ответ

0 голосов
/ 09 мая 2020

Единственный способ - создать словарь:

dictionary = {}
if str=='first':
    dictionary[str] = df
return dictionary

из вашего кода:

 def basic_pivot(df1,df2,str):
    dictionary={}
    if str=='first':
       dictionary['venue_df'] =df1.pivot_table(index=['venue_id','max'],
                                    values=['match','name','continent','country'],
                                    aggfunc={'match':'nunique','name':'last',
                                             'continent':'first','country':'first'})
    else:
       return None
    return dictionary


dictionary['venue_df'].head()
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...