采用python语言对csv文件写入字符串_使用python在CSV文件中搜索字符串并写入结果...

#!/usr/bin/python

import csv

import numpy as np

class search_csv(object):

def __init__(self, infile, outfile):

infile = open(infile, 'rb')

read_infile = [i for i in csv.reader(infile, delimiter='\t', quotechar='"', quoting=csv.QUOTE_MINIMAL)]

self.non_numpy_data = read_infile

self.data = np.array(read_infile, dtype=None)

self.outfile = open(outfile, 'wb')

self.writer_ = csv.writer(self.outfile, delimiter='\t', quotechar='"', quoting=csv.QUOTE_MINIMAL)

def write_to(self, matched_values):

self.writer_.writerows(matched_values)

print ' Matched Values Written '

return True

def searcher(self, items, return_cols=[0,2]): ##// items should be passed as list -> ['OneTouch AT', 'LinkRunner AT', 'AirCheck']

find_these = np.array(items, dtype=None)

matching_y = np.in1d(self.data, find_these).reshape(self.data.shape).nonzero()[0]

matching_data = self.data[matching_y][:,return_cols]

self.write_to(matching_data)

self.outfile.close()

return True

def non_numpy_search(self, items, return_cols=[0,2]):

lst = []

for i in self.non_numpy_data:

for ii in items:

if ii in i:

z = []

for idx in return_cols:

z.append(i[idx])

lst.append(z)

break

self.write_to(lst)

return True

### now use the class ###

SEARCHING_FOR = ['OneTouch AT', 'LinkRunner AT', 'AirCheck']

IN_FILE = 'in_file.csv'

OUT_FILE = 'out_file.csv'

non_numpy_search(IN_FILE, OUT_FILE).non_numpy_search(SEARCHING_FOR)

从你的问题的措辞来看，我认为你只是想完成手头的任务，并不真正关心如何完成。因此，复制并粘贴此文件，并将数据文件用作“in_file”值，将要写入的文件名用作“OUT_file”值。完成后，将要搜索的值放入“搜索”列表。

需要注意的事情。。。。

搜索应该是一个列表。

搜索中的值完全匹配，因此“A”与“A”不匹配。如果你想使用正则表达式或更复杂的东西让我知道。

在函数“non-numpy-search”中有一个“return-cols”参数。默认为第一列和第三列。

如果你没有女儿，告诉我。