代码之家 › 专栏 › 技术社区 › duckman

Python:如何基于函数的参数创建csv列

python-3.x python

duckman · 技术社区 · 5 年前

我有一个函数,可以用一组列写入csv文件。每当我想更改某个列时,我必须进入我的代码并手动更改它。我如何输入列名作为参数,它将改变其余的?

我当前的代码:

def write_to_csv():
    data_writer = csv.DictWriter(open('output.csv', 'w', encoding='utf-8', newline=''), fieldnames=['fund', 'Ticker', 'Price, 'Date'])
    data_writer.writeheader()
    for row in input_file:
        data_writer.writerow({'fund': fund, 'Ticker': row[0].value, 'Price': row[1].value, 'Date': row[2].value)

比方说,与其更改列 Ticker 到 Name 手动,我想在函数中使用一个参数,以便:如果我需要列 股票行情 ,我会执行 write_to_csv(Ticker) .如果我需要名字而不是股票代码,我会执行 write_to_csv(Name) .

我认为我面临的问题是调用引号中的参数

0 回复 | 直到 5 年前

CaffeinatedMike 5 年前

这其实比你想象的要简单。

def write_to_csv(column_variant='Ticker'): 
    # Use a with block so the file auto-closes
    with open('output.csv', 'w', encoding='utf-8', newline='') as csv_file:
        data_writer = csv.DictWriter(csv_file, fieldnames=['fund', column_variant, 'Price, 'Date'])     
        data_writer.writeheader() 
        for row in input_file:             
            data_writer.writerow({'fund': fund, column_variant: row[0].value, 'Price': row[1].value, 'Date': row[2].value)

推荐文章

Google User · Django管理员在`list_display中未显示`creation_date`字段`

6 月前

user29747013 · 如何创建一个新的数据框架,其中包含原始数据框架中列的聚合列?

6 月前

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

6 月前

user29715306 · from_users=和chats=电视节目中的差异

6 月前

Redshoe · 当执行numpy.genfromtxt()时,python是否会读取文件的所有行?

6 月前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

6 月前

prayner · 更新嵌套字典包含列表中的项

6 月前

Bringo Jr · 我可以在O(n)中解决这个问题吗?

6 月前

Dave · 如何在for循环中修改列表值

6 月前

Shukurullox Komiljonov · 从记录中获得相互和解。使用SQL

6 月前