Python提取CSV文件信息

最新推荐文章于 2024-11-01 13:59:45 发布

原创最新推荐文章于 2024-11-01 13:59:45 发布 · 3.3k 阅读

2 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Python 专栏收录该内容

0 篇文章

订阅专栏

本文介绍了一种使用Python从特定CSV文件中筛选并处理数据的方法。重点在于如何合并指定列生成新的视频标识符，并筛选出英语描述的部分。通过示例代码展示了如何读取CSV文件、进行条件筛选以及写入文本文件。

Python3.8

Python 是一种高级、解释型、通用的编程语言，以其简洁易读的语法而闻名，适用于广泛的应用，包括Web开发、数据分析、人工智能和自动化脚本

前几天下载了一个视频数据集（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 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')

最后得到我想要的结果：