搜索热词
下面是编程之家 jb51.cc 通过网络收集整理的代码片段。
编程之家小编现在分享给大家,也给大家做个参考。
# -*- coding:utf-8 -*-
import csv
import os
class csvread():
def __init__(self,filename,lstname):
self.lstname = lstname#lstname 用于比对csv中相同列
self.filename = filename
self.header = self.fields()
self.data = self.getdata()
self.dataset = self.getset()
#test update
#以字典方式获取csv数据
def getdata(self):
data = []
with open(self.filename,newline = "") as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
data.append(row)
return data
#获取序列
def getset(self):
lst = set()
for row in self.data:
lst.add(row.get(self.lstname))
return lst
#获取表头
def fields(self):
with open(self.filename,newline = "") as csvfile:
reader = csv.reader(csvfile)
fields = r