Python for loop on function with args for csv export

Question

我一开始写的这个函数没有任何参数，它 运行 很好，可以将值导出到 csv。然后我向该函数添加了参数，但现在它不会导出到 csv，尽管代码为 returns 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])

Answer 1

不需要循环。您从函数返回一个 Pandas 系列，只需使用它的 to_csv() 方法。

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