使用 Python DictReader 获取特定行和值

Question

我有一个 csv 文件，并试图在第 20 行第 3 列上获取一个特定值。

但到目前为止，我所做的只是在第 3 列上显示所有值（此处称为“名称”） ”）。

这是我的 Python 代码

d = DictReader(r.csv().split('\n'))
for line in d:
    score = line["name"]
    print score

如何仅显示和获取特定行的值？

Answer 1

天真的解决方案：

  target_row = 5
  for num, line in enumerate(d):
      if num == target_row:
          print line["name"]
          break

我删除了中间变量 score。

请注意，这不是最佳的，因为您会迭代直到到达所需的行，但我不知道是否可以随机访问 DictReader 中的行。

Answer 2

向前跳到所需的行，就像在另一个答案中一样，但在 CSV 读取迭代器上进行切片而不是手动循环：（

import csv, itertools
rownum = 20
colname = "name"
line = itertools.islice(d, rownum - 1, rownum).next()
score = line[colname]

我知道，这不是最漂亮的代码，但这只是为了说明这一点。）

Answer 3

简而言之，将 csv.DictReader 的输出转换为列表格式并通过索引访问值。

注意：通过索引访问时返回的值将是一个对象，其键对应于 CSV 文件中的列名

---

用例：

我有类似的需求。我只想读取行的子集，例如总共 100 行中的第 10-15 行。

一个有效的简单解决方案是将其转换为列表：

# Observe that I am converting the output of DictReader into a list format
lines = list(csv.DictReader(open('somefile.csv', 'r')))

# Once I have the list of objects, I can simple access it with range
for i in range(10, 15):
    line = lines[i]
    # If you want to read the value of a specific column
    print line['column_name']

希望这有帮助。

Answer 4

如果您只知道列号和行号，您可以这样做：

TheRow = 2      # Note that row 1 is the header row
TheColumn = 2   # first column == 0
for row in d:
    if d.line_num == TheRow:
        print("row %s: %s" % (str(d.line_num), str(row) ) )
        print( row[d.fieldnames[TheColumn]] )

输出：

row 2: {'t1': '12', 't2': '23', 't3': '31'}
31