Python提取CSV文件信息

来源：互联网发布：中国独立电影知乎编辑：程序博客网时间：2024/05/20 13:16

前几天下载了一个视频数据集（http://www.cs.utexas.edu/users/ml/clamp/videoDescription/）

Collecting Multilingual Parallel Video Descriptions Using Mechanical Turk

其中有个csv文件表示了数据集的一些信息，如图所示：

video名字为：VideoID_Start_End, 每个视频都用多种Language给出了Description

我需要将前三列合并成上述video名称，并筛选出Language=English时的Description

用Python导入csv包是最容易实现的方法。

import csvreader = csv.reader(file('MSR_video.csv','rb'))f=file('MSR_vi.txt', 'w')for line in reader:    if line[6]=='English':        #print(line)        v_name = line[0] + '_' + line[1] + '_' + line[2]        #print(v_name)        v_sent = line[7]        #print(v_sent)        info = [v_name+' '+v_sent+'\n']        f.writelines(info)f.close()print('Finish')

最后得到我想要的结果：

0 0