Python提取CSV文件信息

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

Collecting Multilingual Parallel Video Descriptions Using Mechanical Turk

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

Python提取CSV文件信息_第1张图片

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

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


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

import csv

reader = 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')

最后得到我想要的结果:

Python提取CSV文件信息_第2张图片

你可能感兴趣的:(Python)