如果行包含特定字符串，则定义 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

上面代码的结果如下：

[['一种'， ''， ''， ''， ''， '']，

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

['C'， ''， ''， ''， ''， '']，

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

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

['F'， ''， ''， ''， ''， '']，

['G'， ''， ''， ''， ''， '']]

假设我想更改包含'a'.

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

Answer 1

首先，使用内置的 csv 解析器。

此代码将定义包含“a”的行的 Python 列表：

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