Python编程之CSV文件读取

程序编写使用的是Python2.7.经测试可用。接下来会进一步完善。后续推出窗口程序，基于Tkinter模块制作。欢迎关注。

# -*- coding: utf-8 -*-
'''窗口程序中，输入框中有root,filename,strb,N,Col
root:表示文件的根目录
filename：表示文件名称
strb:待查找字符串
N：带查找字符串所在的列
Col:提取某列数据
'''
import csv
'''------变量定义-------'''
root='C:Desktop\\20180107'
filename='test.csv'
URL=root+'\\'+filename
strb='-1'
N=1
Col=4
'''------常量定义-------'''
strvalue='DataValue'#固定量不做输入窗口
listfile=[]
data1=[]
data2=[]

reader = csv.reader(file(URL))

'''------------------提取csv文件中的某列数据-----------------------'''
#将CSV文件中数据全部存放到listfile列表中去
for rows in reader:
    listfile.append(rows)

#将listfile列表中每一行的第N位字符串和strb相同的，将listfile列表中该行数据存放在data1列表中
for i in range(len(listfile)):
    stra=(listfile[i])[0]
    if stra == strvalue:
#        print listfile[i]
        data1.append(listfile[i])
print('------------------------------分割线-----------------------------')
#提取某列中某一数据所在的行数据
if N >=1:
    strb=' '+strb
    for i in range(len(data1)):
        stra=(data1[i])[N]
        if stra == strb:
            print data1[i]#输出满足的条件的所有行数据
            data2.append(data1[i])
    #提取data列表中Col列数据

    for item in range(len(data2)):
        print((data2[item])[4])#输出满足的条件的所有列数据
else:
    print 'OK'
'''------------------提取csv文件中的某列数据-----------------------'''