【问题标题】:Referencing keys in a Python dictionary for CSV reader为 CSV 阅读器引用 Python 字典中的键
【发布时间】:2016-12-23 01:22:10
【问题描述】:

python 新手,并尝试构建一个简单的 CSV 阅读器来创建现有工具的新交易。理想情况下,我想建立一个字典来简化设置新交易所需的参数(而不是使用 row[1]、[2]、[3] 等,我想用我的标题替换阅读起息日、交易日、价格、数量等)

我在下面创建了字典键,但无法将它们链接到我的脚本以创建新交易。我应该用什么来代替行?任何建议表示赞赏!谢谢...

代码如下:

import acm
import csv

# Opening CSV file
with open('C:\Users\Yina.Huang\Desktop\export\TradeBooking.csv', 'rb') as f:
reader = csv.DictReader(f, delimiter=',')
next(reader, None)

for row in reader:

    # Match column header with column number
    d = {
        row["Trade Time"],
        row["Value Day"],
        row["Acquire Day"],
        row["Instrument"],
        row["Price"],
        row["Quantity"],
        row["Counterparty"],
        row["Acquirer"],
        row["Trader"],
        row["Currency"],
        row["Portfolio"],
        row["Status"]
        }

    NewTrade = acm.FTrade()      
    NewTrade.TradeTime = "8/11/2016 12:00:00 AM"
    NewTrade.ValueDay = "8/13/2016"
    NewTrade.AcquireDay = "8/13/2016"
    NewTrade.Instrument = acm.FInstrument[row["Instrument"]]
    NewTrade.Price = row[4]
    NewTrade.Quantity = row[5]
    NewTrade.Counterparty = acm.FParty[row[6]]
    NewTrade.Acquirer = acm.FParty[row[7]]
    NewTrade.Trader = acm.FUser[row[8]]
    NewTrade.Currency = acm.FCurrency[row[9]]
    NewTrade.Portfolio = acm.FPhysicalPortfolio[row[10]]
    NewTrade.Premium = (int(row[4])*int(row[5]))
    NewTrade.Status = row[11]

    print NewTrade
    NewTrade.Commit()

【问题讨论】:

  • 行如何被数字索引和键引用? Row 是一个字典,所以我不确定 row[4] 将如何工作,因为 DictReader 只会使用字符串字典键。
  • 请注意,编辑后您不再创建字典,而是创建集合。
  • 谢谢——是的,看起来我做到了。我可以用集合和字典来完成同样的事情吗?

标签: python csv dictionary key


【解决方案1】:

csv 模块已经通过csv.DictReader 对象提供了此功能。

with open('C:\Users\Yina.Huang\Desktop\export\TradeBooking.csv', 'rb') as f:    
    reader = csv.DictReader(f)
    for row in reader:

        NewTrade = acm.FTrade()      
        NewTrade.TradeTime = row['Trade Time']
        NewTrade.ValueDay = row['Value Day']
        NewTrade.AcquireDay = row['Aquire Day']
        NewTrade.Instrument = acm.Finstrument[row['Instrument']]
        NewTrade.Price = row['Price']
        NewTrade.Quantity = row['Quantity']
        # etc

来自documentation

创建一个像普通阅读器一样操作但映射 读入字典的信息,其键由可选参数给出 字段名参数。 fieldnames 参数是一个序列,其 元素按顺序与输入数据的字段相关联。 这些元素成为结果字典的键。如果 fieldnames 参数被省略,第一行的值 csvfile 将用作字段名。如果读取的行有更多 字段比字段名序列,剩余的数据被添加为 由 restkey 的值键控的序列。如果读取的行数较少 字段比字段名序列,其余键取值 可选的restval参数。

【讨论】:

  • OP 已经在使用 DictReader 但似乎代码混合了 csv.readercsv.,DictReader 代码
  • @Euan 在编辑之前,OP 没有使用DictReader
  • 对不起@Euan 的混淆 - 这是正确的,我修改了我的代码以包含 DictReader。我正在尝试更新以包含相关字段并收到以下错误消息:KeyError: 'Instrument'
  • @YinaHuang 在您的 csv 中发布前几行,包括标题。这应该可以解决问题。
  • @YinaHuang 重点是,你不需要新建字典,阅读器已经是字典了。但是您确实需要知道字典中的标题。我怀疑“仪器”实际上类似于“仪器”或“仪器”(末尾有一个空格)。
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 2010-10-04
  • 2022-01-02
  • 1970-01-01
  • 1970-01-01
  • 2012-02-28
相关资源
最近更新 更多