python - 如果行包含特定字符串，则定义 CSV 文件的行

Question

我已经学习了两个星期的 Python，在用 Python 读写 CSV 文件时有一些我想知道的东西。

我使用 def 代码输出:

def csv(filename):
  f = open(filename, 'rU') 
  lines = []
  for line in f:
    lines.append(line.rstrip('\n').split(','))
f.close()
return lines

上面代码的结果如下:

[['a', '', '', '', '', ''],

['b', '', '', '', '', ''],

['c', '', '', '', '', ''],

['d', '', '', '', '', ''],

['e', '', '', '', '', ''],

['f', '', '', '', '', ''],

['g', '', '', '', '', '']]

假设我想更改包含 'a' 的第一行的值。

我可以使用哪些代码来定义包含a 的行？

Answer 1

首先，使用built-in csv parser .

此代码将定义包含“a”的 python 行列表:

import csv
rows = csv.reader(open('yourfile.csv', 'rb'), delimiter=' ', quotechar='|')
arows = [row for row in rows if 'a' in row]