时间戳

首页 > 解决方案 > 用于 csv 导出的 args 函数的 Python for 循环

python - 用于 csv 导出的 args 函数的 Python for 循环

问题描述

我一开始没有任何参数就编写了这个函数，它运行良好并且可以将值导出到 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])

标签： pythonpandascsvexport-to-csv

解决方案

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

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

推荐阅读