Python для l oop на функции с аргументами для экспорта CSV - PullRequest
Новая
       19

Python для l oop на функции с аргументами для экспорта CSV

0 голосов
/ 17 апреля 2020

Сначала я написал эту функцию без аргументов, и она работала нормально и могла экспортировать значения в csv. Затем я добавил аргументы в эту функцию, но теперь она не будет экспортироваться в csv, хотя код возвращает 0. Функция может распечатать и добавить список. 

def grp(department, group):
    df = pd.DataFrame(data=students, columns=['Name', 'Department', 'Script'])
    df = df[df.Department == department]
    df['Script'] = 'ADD USER ' + df['Name'] + ' TO ' + group
    return df['Script']


with open('output.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    for val in grp('Biology', 'Science'):
        writer.writerow([val])

1 Ответ

0 голосов
/ 17 апреля 2020

Нет необходимости в l oop. Вы возвращаете серию Pandas из функции, просто используйте метод to_csv(). 

grp('Biology', 'Science').to_csv('output.csv')
...